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


static const batchedGemm::BatchedGemmConfig tllmGenBatchedGemmList[] = {
#ifndef EXCLUDE_SM_100
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "f2d3ccd713cce308e6edb5246341a4ff52f60edc571923941dc8d0ad8a73b738", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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_relu2_bN_rgTma_clmp_sm100f", 256, "ae13449e17ec3c2e4c0d56496966e7fc07118ee6f8334a69bde9d690a5c4c186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "f540949e3865495aebb3ffb88ff40e6f886c87b209e6b253a0905cbf5254ef77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "6a0db174b5ad0073cebb807d225664386f6d34ff01e26223f429063763ce9f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "597663f15597e3a5e0aa07885e614ef79acca742eafc17134d7431cd46aee9e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "2908f9d48f913cbf2d59019c31d4a855ce90a6652cf564c6c43bad4e5ffa33c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "f1638c7a92dd4af3fa40cbbad811fbc634244dbca118edacb99a18efbcfe790f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "ab662ef38916939dd1c9ed6dafada1d8dc709cc24b325359862e5cc51c72fdde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "553c2f4a581c666aa80339509a7093a92b8ab2150af421727dde0da41ec132b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "e2add0fa5f5628cf87ed6f0139b1ab1ac778aafef1551545249dbab662abfc99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "96049a8433d2b842b51aee1a9106e11de28ef0c64333a05ddca4760eb96dcfb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "ce3db07ce569bd9dcf415900da09320dd0c33788913f2eb6771a33241849e863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "52da441a6201fb553b9db53cb9aa18d481fdd896c2b31276c2c990775187466a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "10800c8435af0e00c39408d9e4aa1ac6c7fdc414483e0012eeacd822daaeceec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "4ee9ceaa1497580c05d5d1287645b2834eebf4869fc1a32989e22c82ddcc716b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "ac160843bce6b1075b076a1f4507045fce05c19ee532466abbc7808eaa8cc862", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "b645403c04b52142bd060ff8c487777ddb639241f49e6dc7c267686f81f1ad33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "3b9bb884d43aef4827e11825e0664e9b6106be2d9c50b263a1fa780cab4d36b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "e1b96f55d77d59547c23d6de38f5c4af87687578a88254e8ef9882a7d7186fe4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "4f411150115070c80ff3ff9ad8994564a5ac4dd60c472bd8a75e8d0ae43162ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "fa09b9d512ee3bad93bebe68a7d701279e47c3a4c71b34b02289aa6a1df9a1ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "e4e3c7b21b475f0cad081bc069ab190f5b3adbe35fe7d6ad290022134ae05297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "af6f8c409b0ad57483f3e89ba3115d65a5495be18b9336c37b2cb93177e87a4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "079881ff72481ef2e7dffcb718361b70a4f49e53c0ef1fb957c3a7e2f82666bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "9696efcf5a2f4b66c84bcf832a735d74eb99e5d4a7b035e35d4648b262bd8e63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "e8ccdfff180b8de0191077ccb9f20e2cc26dbc979df46600f86d1ce282bc58c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "57aafb9fac7fd2bc3387e6b80c173fd7f057c501a0c15e815d2e25ec45e9322f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "dcda741dd9dd4895622e651d92d075004f5d14ac513b13b2920a3efbb8e63399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "effedda6b13f8bfd657f803343ed6d6215b92148105113f4ce3d5d5d68cbe51f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "d761e10bc880cc6fa245be901a3e0bd3c4946bfe3223fd5221cfffb3a53293d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "9bb469867da9a23e6f9e3ef2831a91bf4a378aa17fb6c3a8cb90ae3daaf76a91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "41c6b249a31514d82c54198bec7ad773c163406a5b7def7a329d10f8f79d515f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "13d11ccf6d6e24f14ac0887c267606752b99c51cc6f8a49fdf5d6810055c5b43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "0ca1a87846b6ce81582b72243559bbfc22793a421e50d85087e343f55d61cc13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "2c47c7f5d01e6e8eb57c6cb8c3462532cb9274636926dc09b0b7d3a6044969d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "05c71de4df576e3d61db07463243d1b8d3dccb06153e47e419ce554ebb7ef31e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "dc62f8061a2ad0334b9fc8bc2a3fc2061687b19407feeba3b68e976615f30af0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "a239a9c562006565f1308304ab4d2c647daac2777bf1c086026fe5bed0dfb5b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "662f4b7fcbb67dccfac5959c7274d85ee3b987049330f11a04cc2d91151d36ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "1d57eff18772d11a021e0b8d3d0115eda8b9859168b800af2a75d15a52275bec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "025e17e488f197d829af8f214bdd65bb2233270fe2ba36efa26b432eee4c7553", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "4c54d1ee87a46b1f821d50b9f7d6ecfabbaf277768ceedabf538b6d0ce2ed5e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "97c049c2b1dc35cb961151b29e6f8014d05875aa98de94b5567f89241dce082a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "a0d046c432b5be48a0a49a903af35427e4182b5aae1ab1e518c28c4526c2f433", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "54340f0da7ee9c568a3d0ad82bbed490ef2ee81a244130fd4a3226301db872b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "4044ec3b832b179cfbe0c3c8f62300efc1a3fea9565c18a2406423517bf4a6b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "708cf5facd55b21a2a95298e50ee41a1930a60f2bec6bd7d91f7fcb3c74a864e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "bb82c4ad76e723ec7c6c698cb4a1a681bb8e7d01d7c874ee5c869590e623e92b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "8d8806c8e1e20aa9b1cf61b9d93eb7ff70e9f4eb921c12fe57a38e133e2141cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "bb26fcdbf5cbb277f68d43e17b59d348147342b682a10e3b193bb40cba5c25d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "ec3e3e4da5842ee5267c1f87c02bc0da71ce93c36b2009d5c35e450a28a5223e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "62f93c8ddd8a8fd8bd34f351da6d4e42348283410a336d22f09cb73223025036", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "cb22aaeef98a48f70681267843dba8b7d853cd0df6792ab6e4bb9047fd3d9f71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "4529a226c8bc26af67f850be4a26fe00bcbe622d497e1ba7bce40f078702719d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* 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, "7746a806319f61122958fc699af2c1d32cc837aa47c5f1c28604f43a588b60c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8cfdca19d9868208e7744315d444be35c323109636346ffd75f4144ca64aaee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0082b40510ca4d5e7dfaa82803624aad082025e3208bc90052874bcad4415de5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aa88b3943ec759ac269af74cd0ec8f5ca5ca550911e5c7fe1992c6280a9729a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e0dea6960fb1570355fca171f9ea28ee50f0b6edf66d63750f75463a882c085", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5dda49980c7fd26e4250ed6e8ddb98153182d79a4ec2caf3874405f3867d2fd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64f8d441623f78580f1ba8bd0b08032192dee2ec5ce503fe0be4b1b5b7248377", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a6b5af2e8947f785b94733800be53c6a7dcd9bf5f97920881d7cb94b175caec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a377a94db3f07251aeb9217205f49da218201fe0fb0944bc336d910cdaab87a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cda853d5bb13c204c947d5a7bf78594f60d6be6b517cf8c4e8a3ba17cc58a8eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "708e84dd068c4b6cac1495370e8ae6e7100e743ce1e2ca1caed838abc1a70df0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "197145d7e8769e9addcacb6e1f42b4cc4a37aa360761d528da8c4902057ff3ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "9ff29cd956c1861b241f0111932348676568b94627fe4ec7d49b53505de557a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "8e048e637fa3ba2e7b136937153e16fd80febb1b91869cd71d2c9a01525eea53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "4932e6460b0927cf563ee5fabdda17a35fb1fe430b1527031f767a62bb683d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "46f987bb33642b9abf36c231e551c2a01b94184305842bf974da7ae4fbe32d8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "c632bd19546531eedc08d133af7dbcc8680a14fb9905917e3a457764f315a066", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "f881a3c7a70916ee6de6b9d9d13c16f4446b8adf0bb66da60f87d5d1ee0d9f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4fec0741d455043f7e29275e02574fba802f1ba820ee5045ad017d0fdc6f56b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "1b8978b146b8ca63b962b0abc6a1fec5a33e125b6c8f2e4f2b616ae5b15f0e75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "03daf0991c3cf4411cb17b00f7ba5d77cadbd8802a09bbb3f55b30646b281147", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7eb0a85cc93b5c504aac59868f0903e4713b6a95de24f78cc9f40cc6c389818a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8ef5bbcdafbc3403878bfd45a22926b515151a09b5cb370a65e84deac949dd3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "00485b6d5f0f1d5cd94659ed76aebce5cde9faafdaacd750e6032d4779b92e83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "39b805b52c75c53b409c328e40fb75a62202c9bf942aab38f847a55c64366b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "ba47104d748deadd5aec91fbfa94edfb400518b14fed49877ed678df55caee95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2306b8490cc5253d25ae3f1a73eb810616bf901e87f9017275b6f4dd0045c2ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "4741b17db7adf2695804ec7c0718c5cab07bcedde9e7f1491d44f74f0442df4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "b2cd1ad634206a05221384b9c58adc33416ddddc6a08221442bf64e5d1da5005", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "7368de58d37ff4c0dc08a40c7996815c2631a5673703a3b0ff891f76b14e9429", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "632414b2770fbd955c8a60088ecfbdb9d39391f67d2c01ebad1aefcbf6bd5d28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60a4a1fa5e6f732bebd78e020fd969c46cde51099c0e39ef45056f94ca3cdaa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "139f00f73f6f9595e18de4986f8fe22fde3f78025255f7f8d14c606ce8c6d26f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f0773a800e705d1722ddd75c79a85c523f76e5045715a56316a6145fa6aed3ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "5188749f316793cfbe159b0860ab2ce5a7466b6cdd4ec57fba31425c403428f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4fcb3ad8308985d34923859201bee6eddbf1d0c1d668d399630f7a073f6fb3b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1f07168794aa7efc1aee1fde30f0651b22c97d65537a755f6f9a1627edd9c325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "a3e9995f08ab091fc03dc9a23a407c47ca6e5ff596295d96f415e59fbb52b452", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cc0ada2ecb065bcd1fda4ab833c4471a85ecc55c082deb7621a795ad3426b46a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "902d1c7792685ba52ca642d4e1eba0fb219f7ef420e2cdab555cd0f6b2f88a64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "c6886ee2c4b7718c1e6848218b9ea2fbc1f69aacd8639f971b894847f911b134", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aae9de52c8d28ee35f96fad5db9a2bc677571b0eb5a3e68e1c05ad1c24d9f11c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95a9a92e33689b20859088bd3e4255e46033d0bf5f4278fbbdc26ca2f6aa6e28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d8ca32b8085c1391f5f43c8b66dff077e23d4c09e77767bce5f77d4b55540d9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db012b8d713ed6924c5080c64c209dd377b9eb359b157e52b0d5fc4ac63de3ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e636c0c543f7d21a9fff9aae6bac7a82ce54b46a5ff2ab775f74ff12c808850", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10f66a57e9f004b8337beb58a9f7c8b8a828f8591c5a7dbd4a49d18f07c5185d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cfb8197135b5d1c92b50fd09184e32b4f081bc8575c6a470a176346b8683be55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "65b04067e87df2f332d0f5bce64b06a47217f585e09fd37ca58d5fb3100fe563", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "589476cbce72944a43b630b70ca46284f5e056e3e630217d40f7d931d119797a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "66f725f9ba5482bcee13ae2cc055d9d1392a86044720848cf947d46f5f4cc2a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "078876918175d830d793f186a267a6924f902132a61fbbdf8f71e9e7d2b2bacc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9b9aa1698b3ab07a0b0ca2a8dcf57bc906cf126f212fc95ad89859c40a4489c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "034f6ff77d2037587fb4519bce20a973552b4edf23c6d7557ecb8724cb92bd3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60114c0d9bb3bf3b6c2867c97690e2b1dbd5532ef972c535221cd98493199091", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a1e13bf44ac1913f4b7334923e5d45351655f2b33d4b7e96d5d7414c5163d5b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51cc29c9efd0971ed694c8b8a9c75ff6b18d22dc7ff21337698a6a79699daca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ee99c937de0babbd0cbf81d1b6e0578698d0cc69a76d4b778b84f6000b6702e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d89737e49e15f679c6e46e6cff9839fb951c9dc8f38c27c91bbd81718e34f623", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ce519f26532f79da175942ca4e582176b646ffdcdd8d730b476904b9bf0b4dce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bdf80078fde830801659fab0f83b65522fe008cb9d1c3adc300ff3074addda44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "195705a574baac0eff9ecf75bcdd5f4729670f4f8ed5eb0b37bc6f0c76b20dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "73e0e75c5165d92e019ac7102d4f5b7fca8dd7974ef820439eda777b84c665da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7440f7d94ebc685516211a56f0c2292aa8d72fe813e7eb5212cf2d0de0798880", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c9667addfb86ae761ce34f59cb527595995c03b3bb0ee1c716c2d60539922bd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe13e951ceeab5c73b8ca69cc1b369013f8a58b9e28a26d5cfb7c02bcf840767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5017461760b22394ed546e0240929aa5933a6738f9d01b4d6319005664f61108", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "50354135c80297e8b53510a0e278b3388f4e23da34f6ae74c3ea9d0c84626322", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "497493c67c6250e6fe864f36a0b9525ad501360c88fa1ae9b8eef0e8e30d5013", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5b8d6d89533e86d67f13f8377d36707602fa11567d40a058957a908d69e3e9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cb405ab2bdc33863d27471f52c6ef10d469e455040ff5f0c1c0bcec2e0d50970", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7cfbc1bf95e799be76881b56e5b729580b868c38ed08bd9f75303746843428b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f0bf0319c79a476a91550e0a67a54f86be2f6ee33f813f09ef131baf3900f299", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "10c6be2790dd120ab8a5140c944e7a61f7eeafaa60f215e9f9ccd74ea28e51d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2b74d7287c40fb97539c368d167c2c75f0e769e7c4b191502784ae9fb60fea02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7c4a33d969ea8c80d0544887e50d01561dee5775aae560bca21c730972d55771", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "efad0aa5e31726931d087b623efc8647d3f5db4c327e9a5888427071dc0f7a35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "d224f033108f7e7b5b0803e5d288092c1e37f2ac7edafd89c6847c51a7663cfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "03c8b3bca2128bfa67a7bf7978e4a137e41b2763f4a7db4774498c9b879442b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ef223d35b0573fda98a22756fd1b6ba526ab364097440baee0f934428d39b820", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "cc1525f1599a0afe231ab1dc59b7fff45c52f4e24e066b2a9328e833f7b9bc83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a1fcd865b48072d54c438b8719313f818959d4ab7f9b7ea6cb3287d5a90844a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6670d1de647cf6f0a894ae415da03fb3faf04f6bdd9f40d424d8fada580505a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "e4b9c30642b3993621d4d0c29e98b9e7301211a7d3b1eaf50f739a18f564fff9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ae81de353510f29df5b772da608eef4e31a07cbc4d913306c5af91d0e8167414", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a463d700e3051a4f527b8b9559b6e470ab58e6df4b41c4f525a855f42cb931f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "5ddf77b965599d9d8d0afd5c0e964cba285b7c40931e220b7e543848ac81c6eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f8b4024512ebb6530169e792cdeb5d31f65f072d6b369af2aabab0d379af4cce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a1be7b66d8f3e920c63e57b5a4fee9c8039cfed6e8abc9f351098384e425f1e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "74a0f965f471cf132e66308bff03c3aa3b3c5c4673ba6e8f2b99f19c022b5bbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a7fe1109803ab317499032ef03b5b63e63e4de4d1e5dbc6c9db00522c23e4cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d9b85a3a2f806c2418f71923778df8751510d338c6673f8075225617be89d362", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* 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, "6078b3cb39e3fd899fa070f7d471c053553d2c62f846308f97deff8c4a8d7810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f019d9affb92296cdac740c4b25e7c3065ac8f0b37dd74316ddf55dd20abb887", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "18fc087d0d9c6c12aa845769b527d16760680d384258c4ccac60eb1a4ed40b6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c7ab8140c566c5fe460e27a01f24c126f04129b309c01c89e4b4a54242a6f45c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "91a540790681c9582fa44f1d42e3365b557cd7142963e70ab7ab95856fd7eaa0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "cd9400d8c92188a449ee53acf83db5ae058aedacd05d924730bfcaf5fa7db8f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "eba61524767b5caa3f4442913f654f674b711bfbfb411babf433d7dc7861ffa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3f5d94d6548caa33b59f0fc4ca54d219f2fc753943e2a2670aa229869322e628", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* 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, "59f003b2257835a1d4e0f628c8eea3461776ef0c974d72db7536c256da0c7c0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ad2e515a3efa3bde6df63a53c2f67fb9f7262e59a801b13340389858533b1c0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "40863a170c402aa48b2a31ac98de8c9407be828cc2662113283517d9277514fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 768, "b6fda959160ff537802b4f0d515a29d89d25d7fe21368dd488b560be349d2184", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "40f44bf07438514ebc4541b8a818069d9841fd1b073507e7d6183e2adaca87ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* 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, "c1f3fbf72fad663daf72ea93175e62c186778b3099de41563609c77a290d6f41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "65d616ff82d2be6b3026edbe662a26fa39b76465284ae7069b044c6805904bb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5143a18e5bbfb36d2960c47e5289a3684446eec81bee397a923aaf89e1f508f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26016f3d7d49668eae2b9f7be065ff25e4657208304c0b46c0447ad843239dd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 512, "b95c3accef8c3f9ddc69d824c788750393a9d9e0133644b095146f45ae2f20e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 640, "2eee3fddd37d875d02788adfe7412388904f564c90db71f5f741858c8f91a7bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ad35ee945ccb6b6e94a6f787043e7ff2d4bd5d5a4c8ce41379b99499de0b2f10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0bf1424f39b2a36b8933a4840377e536d138575a1a0dccfba10a337def1d49ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eede0a72f63631e1980082a3b5e29b8277f16074cd826472b478799b2ca07ee5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5de0c17c9a05150df8aab029fcdd0f9c804b2ec3bc5369d28b704590627ade0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d09b81a75f86534f16c72039f7dca5d76b9a147789f9a368cbb0e77e6359bebe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ccc2cc0d03564e28ba04e51342519a7d5cb98fd429b14b117ec5b9f3f13c07ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "08caacb9edfe3680d7802f4c82bdd6f04166e9f0fc237dc0318ab42117dcba06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 768, "eaf4ee2625ba1b18376703019bc8fea484d84437a8eda9980ca6d5852ec2e12b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48e4ccee104ea773db81398f1cd1a23f5929b84c8b901aa86621c33a27530e5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "7b60b378f0f8a29ae1a7b07de32ec577cf8ef42d2f35d8fe554f0547b97a2e99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0e801bdf71bbe2663c0e586f9ac3fc107dd1b9d636a8584b02783c6198f43fe7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "2cd599f8bf34e05901351d46d6fd77549efc7081b98e380a224ac3e73e9be481", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f56b15dc640b348d31925f526dd3231b598977ae26f56deb1f85364405b5f31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fedfb08bfe761ec3df46093eac0a94e9446dc8f5e9f9b210029abadc3ba046c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "397ef955cb31d09ecf737c349166ad7bfef42e5bc361c38a1296a997cd19d29a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "48b7dcaf7aaa9ee8b3fd0f746037e49973f9321d38d682506b2b4186cf5c2048", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "dec59316f9bf9ca9269cc3952efbdcf6fbd42f0693a0cc5c5b665a8fde58e090", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d31054b4d09ed5ae65910e5c332aaabf143236cd7c5f396cad0c2bb35ffd93a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "07ffd010780abe66187daf1d6d20eda7f1ea6a74dc147d1d198a4c22d193d1ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "6d2b6577f4aa977f756bf5c86fd4a23b4bc075ada7eb3220ab34b5c39c2dcb2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b314eba2fc8713c98bad35dccfb4cfd4220d0c4d202738f73b2d03fdf9acc198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "38a9d7623f69e31736b9314f0c1a98798af728e63f0ad4f9fdcd0a95714d53a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "cf14296e36ae293933f37d5e90bab323cff67cbb5b9adc6830b157d8080a4f7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "562668c956378f55593d0fc316dc38fbf2bb210b4348b46a187cbbd962b34ca2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "6d443fff1ece394293c5bd30ec0834eecc96db77a7e6a751391d42ac1ef4e2a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b1e5c3ead6ec9b3c1d4f672b8a327ba09cf9e88fc97dab9c83b363ad2eace250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42fceb249e77cd52639213ed17baa83dd47ab245105123dcf6d97fc7f869c997", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "ac354e78489bfa737b1c80c9360983bf37e76f699bf5fbeaea10a84bfe996624", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "53973fc96cf9cf0d52ae83acabc3a0569ee33fa8245c55c9ee47ca77a5f3b361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7dcd691d9c77622cc919f706b11b7ad1fe30ef35c92c64e81014d582b7bb573d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "89345bc4cd1aa8c1006eb7dd483020142d81e6398baef75a592adeee7315b592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "fceecd65aa4422f4bfde25aeb4ec471a890404924f068d5cb1cbf0f888b6a75a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "bf7f8893ad68cc4d554be47c0b0001e232d80a1931f5aa16ae42a5e5d5b1e629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ac3be74f6964546d5a6798f48d6e0b9667777298f399599932154d67d929f17c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d1b3e94118f6a966d9cd51c57da194012765b3eb7539ba24a377dbf4c62ec9b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "1a09400c48d38becffd424574a7157b7c6c6f6f54f75e1230b8ffa87b45f7b6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5da8edbbe0f6bdf107f876c0613346836f493e0654f91f09e781c3b4764f3411", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ca39dc959018377fa6ec352d0bb9e1e656d6a3f41856b5ec9f5a02fb97e84b14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, "b4394439c72dbeb87b0806ef8b5c96860c933c6f1b3eb91c16da965fe9bccc55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "73db1d1d6e02e247e6d2436301e1b94f1868a6abc6416f966b656c16eb7fc09d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, "8191f3b448507fb914d7917ade2c42e50ab7dfa986f34b40746b25c8f63141f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "155164e4b49285461323ad3a66a49a927f7266775d024e1f18c0281cfc68bb9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d4be2486b078c13aedb57bfa79b9290e98ab2aa9de3bae600f02e8dc01d89a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "834f064fe2c0efb24ff5523a6955a86c190a5902390f7acef9c295bc1f4b3987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "435ccfca88fffcec22c38942c755c533cf89d3976242805268ba9e95a38c35df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a4b7bb99111d43041d978dbe613582a445f6839f1397592083640fa515071603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "173aa8dcaff13b369abcab12b6ea07ec31bc56ea1092ce11823039711ef11629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 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, "3f9c007695e95acc58e9989eb225047126b86e425331fb20a3a2814c4f488951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ded3564d2328b3f8919728c4c30b675eaedd0c246a5f59c6f860cf4f709e3c0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f661ce99d18266dfd9126590d84f5a0d700bdbc899a450cba9757d1b94e91eb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* 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, 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, "1a96a62d84ccf42f1248199376a5a07cd306e66a40eab59eacfec86a60d9b7f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "fd7ee0eebdec4082d00f1dc6f9a3aa0ce9e1f1b1671a5014ba0d7389d6842461", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48a2245116e680cfde356582fa4e20ff403fa4e700f12ad375fdc29c5d3e6835", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "586941f4b81cb25e8af1468b041f31895f8333d23d6d89c4d48eaaa76350255f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "5c81681c5432c849187874ade7b44bda81a2134956365df615b81d250cbdee48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, 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, "514e506e566d3387554bb28cc076c1c888685f86afcfa43603f84e0e12e518a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dc76fcdfd9134038e31c500123924fe266e54898082132e7677958fc093b72c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f864cfaa17003ea5da2d0b4ab4d590dd3ce6c244a1e83125458feacc7de69fb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64badee5dc0ea239c857bce5040eba2bcabde634140101a8f5a232b6b9bf8fe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* 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, "51f433bccd10f655023b35e65ec2dae52d5b6e5d4387a3d26cb82c6815bbe4d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fcb68e7725605d1e5e6407e677904f75e59b34ccb24676dc37346046c22be811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6df2c8a5cee5dd21c1d67886d6d5febe7c91e24bdf8e1357bdab0a456c2e3da2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4f8af41cfc4d5ba2c8fa5d039e42394889d6e35e4376bd13047ad322fec50e84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5a545952b3651315bd35f1323e323f8db64ebcca518b4f7f5d38b8ab3a9ad918", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f098dd8836086808f06e8f2397b4fbd8f0b88dbe85f51954917e18eaed03cfe6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "53c803b9a09d08c368f1b76216bb45e9a2094890ad4ee4474ea43b1784c9ca5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9843e0b3e61d9f0e49af2f7114abbe3c9c295f7391e0304baac24bfa7c75eaee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "19118256155f404c6b0a9dbf558e03009c46cff8a8bf88040ba60bb481952c15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39494f40a59024e56d5ba5fbc4c313161999d52ccd3fe831ab2ead73be03e601", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c083b2a9b7a5a8d610b9036bcf6ef398a332e577c170f67ce3e032d4ebeaa948", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a1297f7b2748d77857daacdff48a880b820b58f32ea6e2957bd2e5c88cb33db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49bc9a6a21a565844af263daed0c3d38bb1149b43756eb727aeb50ed7d352186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "227f938c35b2c739a0f49a14e24cca0fe34afa9bc341c5780e3c26967af81b37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d15c39abc1dfc2d1e5e02db8fbb4d5186a79bdf631b3f94321aacadfd74bef2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6752ae091b1a129879daf4a13fe5bf860f9352c2a05b070348adda2805dc88cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b992b4e7ca0d536d78eea3ec304b77272bd93a8f80039c91f3d5cb62566d202", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5d7acc0ac453831de6e0a5898a50037e950c2db7396492909268bb0b368a5ca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "b997690742a75884c79a39c53b943b07a2a1651115f722f856360411d6202a98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec7ede40e7b3500b9d66661721ebf3239a8d10fe182403c296cdae0ff84c62ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6211635be9ea486733e3050b1d350873206d13037dff4e33fef043853012e25a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c47669c5e71e48712b1d46f32a9f63aaf00970e40a1b5657191b0cdf167a75fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2b8262567fc798bc1aada8337955ef31866131a50b5278e8b8d167cbb4314231", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fc3e32b141eee06a759eaa0174db71f6f64dc8ce051515b6c904683183e6c378", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b1cceedb67298f9316bbc37948256a6ad9b1d7b6df3fd76f87fc0eede48028f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a05b91b2699d1a1326ea0ded131f45619bee727a058c215fbbb686cb351e8afc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b221dd7a34d1ba22f7bf32d51a239e21ca19640b7a6128a4e422ff6b1402f4da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "76181a7bf7544e7faab828b70f1a710a6dc84f48ea2171a5e2a6a1b070a2d2b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fb13a41e6d87f23da68174a749fba328e923f43785422cf2f2251a0f1f6e5ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "6924c1ad919a4b2294395cccb930951958e793327b771323e3ea9eb32636d7ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "26719f11693709c24e4e4811f2a5465cd24de2fb1f0b1e98d2d7149505f195f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "952065eb3ae03f73fce9a6d2658d8d658271a3dbabe026750e93cefa4565d48c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ef98d3c12ed39ae5ad298f2369ddb6cfd7e047c17b33eb8a7b88ac7dfe5ca9a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6205300ab6ab4d2a7b0e4c56d34fccf25bf75062a36a163621d18e2510c4a218", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "453985a09b0b918d44c388887c65dee3a462de5ca06a9a78d9af43474f6dc15e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2d1f08ba4376d9cc4ae7fd4bbd410b8c426302595a2136f7bb9f0d964863d4ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "88e25c78cb50bbded1612ca98542e8ae6a045945e8824e9482d2b9cbc8aa3a12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "02572c186c46da183ed942313c5846b52a6eefd166d1c19b8371ffa9e92efae3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c18f361b537189ddfd39f941a4ea3839f58b9aecf31c62543141f0defced6b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "704316211d78abc5a3c91eaa213afa69259dbc7fe4b23243ea3a2f9a3d375bab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a7150443805f0caa43102e33baf82487c926e77ec2f20b4cfba416ef08effdbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7b0437632796c7732a6cb3cdd2b19bdf39adb99a4772696f43ed737d003eac80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1eb47979e556e2c8542a107c77d48e2763f74d1f0e412ba5cdc722338495e99a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a6113ffdc648d5b01f110e79f76beb2d7a02118ead75072348061dd18d184722", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "f094e59e6a3bfa8ce3f0cc0a9ae124043209605b491799122d9e601cfef5d736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6f2d697b7c0da5da08c1b36201aad420d9637e6914be040d6f7b0c1bb1d5614e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "c46b3c280b362c2b2e2bd5331dc85d489868c30784ff5cc482475bc3dd220757", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "604fa9b9d5b57a7870cd49864375be3b5b5165a8e6b8e7c3b49528bf42d62ff4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1983f716f58c9c8d24dde6bce40423be822130c2a55d9217ba8dcb71fc890252", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f0e34d90969141629e87aec85047ed967cd17a830a963f928b1c9285143b128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e04498714fde6102f6e6748d5a56edd69fa8aa6014e3f84d391651919ab8059b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "96ea03e443803d43a691157d7e50ff52038483948e734a3ebf0ed3c27d61bf92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "de0182353c757711f9af5d42522931b0e2dfb5c53d695fb425111d840a41c5dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "72ec2f44d7daa9dd5b858168d6aae334a8f4e95612ee74d949264878a8caf700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb708dcc78e331575527cfaca56316e107c4257995d52e7c098662b4df2ba19a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c569ec72553b2337a146789888ae984c87275584bb41807f0aff66b8c3c4b0a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76ade785eb85cf48d84aa5f6096f22e3e5df1c20c9ead63fd950d7e048a4d6e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "54b5da947818e513513c501744d3c056ada47d82b453d4670ef2dc96e540be6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da08b27814a707b4b5281df30eee67e6d6794116376d82fc5675aee028d65547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2558544e0f92e796daea72135eb816d61c2f21da86139ecd70bcdeff18b60502", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 512, "30509d745c997f97f477dc0588386b044fe30431512645e122d604c13a5cd913", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6fd7ead02ddc88b12826ffa6a8ddb5f3ca42e65fb2e4ec56459a10995e61ed81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a402dcb79999e9eb2e04fa2c5d90f2a36c5339315871eb0640bd11f63dda77bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82237282edf31c037f295d080530a4dad183bc7cae442131027ae0a422d03621", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "749fecf80a766fbbb3d728776ac7408d0da4a7e54c6cfb2b3301c25752ce3907", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_swiGlu_dynB_sm100f", 640, "9f474c93970cb48365a625ef98e0f9823a7a2cabafa964356dd1e61926f2cfe1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9abb27a00a35ffc6111abe563ad70336db1120ba3e50a39e18ae616fff2c96e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8fa842892d939dfadc5869abd98034e74e7cfd14618f2faf6631a1e99c246f45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "002f03a7657c499dbc25cf61347e63b1bad1a34692314548cf25d163733f92cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "ee0e5ce399b2bd75c87c654354f244fa5729f2445c4cc841d39b8d40e2fa59ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "fdac8be363a22f57c09005be2c238e222d63e4bad8604199bbc30cc9a0c11a73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "c75d4e1d08917ed7411eeda26aa5af9cbfd8e211f0f931acd7f0d3f06f7e4685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "631ac9344f86464ae29427499ca48cb283c853a05cf62d57678a0c07185f677d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7dd931d8c0a883e21ff8d41d1f3a29a49c42adeb5ee04b937bd089c75c674bf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0d5009ddf2e587e1509b0b9e3647ec80303a25f42641ea6a076f0591f09978a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f3f6748235a4a664d50a5c068809cedb1a0335d033855ae8973793e104a1c85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "ff47bda7b7fd7dfbc5a50d259d577a93c5dce642ce81adda7109089d4d6ea943", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "22866342aa0ce69dd3e962fa8475398bfb3ddb51e090c70e0645f026af8c96ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9abb12c20b7bc264df711f2e14ba8b893873002a2ccd58373643c3e53a76f643", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf0230b59e1a93773f4b26d1be45db47261b738ec01f71df93faad4ea9dca0b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0a440b7532333312eff09a9d6ce54aae31ff00615051749e15d6043c977d8a7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "4287a6e0bcff043fb5d8408c24d794465a92dd5a753aba41543cc6fd21325f4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "7b084155513507970bfb3f9948c5effa4bcd98953fadc9d95def61d3015587a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9b5ee968a078d96f526a0f609af2d73d06797f6d019f6f539f01835ce218fdba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "055158eb7a379579468ca28d04c78ce2374bc858eea5143a8a0e99604b3a1eb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "bdf2b22057be5e07a27aa120c1cd80b8c2f54568fdf770f0a9c3c5d1bf0ba0a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "460b254c68f2abd4d09609728b85a5ca771880723dce9e0a8e65164523827925", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "2c0ca9ce6070780ad53d312b4056c626197a35ee6d4b81bc48268ff9ef2fdb93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6458eada80599479b7645260dd999bd319927f533695b04f0104f09fe6f2e5d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26c1e9e443f8a630170cedfeaddaee895777299810890562713b1be3533837e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "49e06b6975ca4e860d32c8cfc2b8b2fe8a8964133a9f155b99e9af9f1afa2f6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "dee1b502ec42a7853b97375981c351f31b5801292456a0bf65abe0998b2debd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b538b83accf670127a2f0bd90063c513239fc8eb4fd3e82baf254e853d632a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_geGlu_dynB_sm100f", 768, "04450bed7f38053bd9f84ecb9fefd2f324cb3b5f90d10a5b363534716c0a1bf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "3d3197ce3bcd8c8aa550c4cd3cf7adf55e38398ca9132069b1bbfbef8a57eea1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "530619d1ecc47400a4599a59471465c22505f9d83d92fae20f3f9bf4ccea6067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55eee3a583c4ede64391b11a59c41060607854076708a51bbc67052b3214a364", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5425f4558268bbff63b86f33a8092a874c4eba22c03b220db41273c0e2450899", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d2961825faa67d35c06f5b909e1ef006cc68d45fcbbf5c9a81ada1e0cbe28a91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a75e9156cd610f917faeb83265ddf091993adfa3321e510de1aeb1ff7214e9cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "73d4673b1a74b1d839e0682582f5a5b2f0ac79f9b0a81afe8e1ac6fe1c9bfbe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* 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, "b6ca59b35e035dfc314859a1120104a3bbfacc5cdb6c6e09dd72d806618778ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "770ce84444d1ad2ed616d948032ba507bd17b3d090827b593c9ceb068f404325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "526761005ddac4a20b35b80979dbff257e05f551c15d3ebe8645a788d7a5416e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2cfc106a9953751a6512466a59f5f71d52396b896477c4e7a3b6ea6c9a0d6869", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2b927cc5dfe737811cdd5f9d4f6dea65556ec836a21a1c561aa9f5c3deeeeeaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "35b5e9e10680eea8c38767a473e00783fa09576214847432da1b8a4a49edcb34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c534a7831b4e1780aeba4562832e79d42074bf37059124be7cb7bede5df07ba6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f4062a5eb544275960b77d8df0d2bbf63cfd142ff7fd0c0b5727367f6f4d20b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "892d966a967744da2587825127621c645cf51b80b5f73120170b3237f2b50b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2d9808124dc2ac9b56be4e1397e589a233357045970e3cf4ac1c8de16b8256c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "32815993d5eec584e3d95dd8ad847de97e2f47f0cfe580f629bc370f95cac242", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* 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, "3edff98b43349e2838e23f82677325157973991698ac01090715e6cdf7707edf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* 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, "0cb345bf559bdd5267c0ece53599c24bce211725c7644fb5e092f411e36b02d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "378b62b8ea370aa66e12ac0597440b1b966128d80733625a003e2645641a395f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "872872fab8c67723c6cf7c9d831b07a41d901f670aac212fb616e297480796b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "eccd300c487b9f5e98aaa9497f99c6d688a1804376da998e8c35173bc3222714", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "44d33c4fecaa1a193793cec790992cd8e317ef783007bd92897e1feeafd3d287", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "55d9fd49fe2c1151fa0ba97055821ea0bc70595c539a4ed0da12dd318ae3df7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bbc516cdc473fa3c5a6a10f1050290519dffefe841a4f988abffc96ee5742a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* 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, "54aecbb9bec267b898aa1e7cd3756b5cc4bcdbf5a5edca0451d73719af730ba0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "899a6ca292d41aae9404a3a370dab0ab1af1ce11b7ce9d5053c7ca4690c83972", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "53cfc3b20b0a7e9b807dfa13b5e478ad3d3ddef7bbcb0f92d83a778649f062d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "a378e607b92e2c52d945616daa94d6cea12d5109a8b2026ec6851d70c421965d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "aa31797b70f24801d3b5a47a313fa652fb28682e457fb4ffd1491a0720a190a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "ba22076fc27a7675dcd56b3cb25c8f5639859568fe4c1bd2ab795212fa67020a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "446471c533b0d1c1152c78460dc7c7f4d33b64c4cfdb6731b0668c7769c5391d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "172a975e6d09a2fa91ce6ee0cfbd6114d5b8811c4acd4816ef844b09daf30033", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "af9b063ec99a9213e66edf54e3d8978a92bf2c302120c8a81a8821c3c8e30aeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b8a0e8e4ae1cfdc2c5f3a0babd3a34c94abb0156bdf76d365d5f1abddd355565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "1fa82a2676d158a8afd79b76416005c6686ab498416bdc6ef74a550ae1fd6e5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1284032f6c9ff2557c4ce4b85f12d2d5eba5ea2f41bb11a8d1721b93468d6324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "0a7676c21187d5745f463ea27737ed7945107a0eb35c45809c278e757028f5c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "926089a027234d3bd863fd4c89e34ec1d096f1121bcfb8b83da7087c823df8c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b18891e253cb0409cfe81073faf662dfd6c2af805813586770ece8f1a015aed4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "f9b735c4741b316237399b2d563919d70b6cef54d01efdc730d0b0facbfedc63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_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, "214555f1b3722c162c92a8277cfec8629733066619356d34f3cd837bb0810d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "e2fdf6e7aa7bc841e1edb9d66abb6f1da50dcbe9db8e062ac6c86268f78831f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "4ad4a9ec042b98007a2e3e2d12ba82d3de4eb467bc1d94c209170d5dd91fa7f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "38493efca5662c6fe6b238a6adcfaebee373be6a72f9d263c610ce876199cd86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "7ae3822a4c94fc473dab9f21172d8e33343717c7a1acd3665725fff9d0391feb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "9195f8ad9827c87d24a5b6f4924322c704e81ac8277f66ce146d3f6686613544", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "e37933ad38dbdca605c2816e3f8b2f06e4e2d57ee99ddebe30c8e955b25b5abe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "50802835de3e43c050da2dfee19de903f81e52cb1273f25a795fc017abdfdfc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "47dac718336d74d85a3c4b54b1fba808a2570689c2c10b0d5422a37641a0c1e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1ef6988b5195a0bb6efa41102b85ce47216fb564108d746e4dc459e2e3f9159a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "82603610af25262fe5c95e879d6e0058e0c39deee7d27b2e6b7062dbf69b5060", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "eb475a5ee32043e06e0ee70110fed6e09c7e02c41e3a793a6314d998d3c976f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "40331bf40bd205bf2cf85a58040311a77a15495153bd02298e227c93320a4d0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "ebaeb581f819df61bfc2653c8df2554af8f21897206dde46863e0d129dbf9987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "47c03fe5661ca7d38ce765874a6aa42e57668a1b97e1b6349352bea882558838", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "94fffb5974b142fd471a38740f05a28f0ec5b487498ac0a8fcc99f69ee3c4912", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "b0267bf9e68777f67ff0d8f97d688bd99e22c3d14bfb48b532cd5d396f0ce47a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "286dc521d5db3fffd11f90106336308a387f12f370bae32151e6f69b8d23b9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0242a05e4070b86c65cc5c720abb7df803727ec27f3a32bd9da41543e5834915", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5918eb2c258a145d25a997d68cfb45d1cbc90f65b7c84261b7f0c209601d7138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 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, 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, "de866135ae21be12bb548fbbc83ed1c81edf8e00a48aa3ec27ba0e24ab9ec5b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fcfe1b7779ad41d65fba4f2d4de6a6bc0e65cd5a3c7d060507e4aa0cf79b54b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "66735932b023c0671b89ae0e342ff9df80907e00de4610fa57a9edd17314dfc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "bcd43d4fec7c2bfaed01b7fe66bcdfe02a86a4322930cabe60d854e2e5406bcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 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, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5bedafa34be54cbb8c69a392dee022a3dd9bd47e25afd5e600241a0accadd38e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7435dc9cc8850e840fb8a6db1c837ba6111492a7910b8230950536ff7fe40f15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5166b1db7fd12341afd029d753cba184b48b0ba78db26ec903f9400de3f810c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "bd5f5d4d1fbc60551d0698c62b20da22a9966e859ee13dad5f1b6f8663cfcdea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 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_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "549504686f2d7a10800217857f6ee1bf0c43e728ec4f316d1eae42e25456c89b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 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, 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, "ccdf20bb394a38238b66d9f77fbdf1ef45a16043f347181fca4521f9c6690c89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "333a35a9428a459d0f0f490e1377b4e58d1af115e23f456d1eb3eb05afb47b02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_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, "dd14401f15b407d31c0de96407fe2707ad5aab3d14ae6536f4bc5b2a6833339c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "8381f4614bf6145e407ef0d00b538f2d7287440ad99f4209abddf063f0d1518b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "09947838a130c79c853947337841d4e81c9ad1072234852c6780e57e695b8628", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6d1da29fe83b481374b8a8a1b8b10e9e1ca8e1c411d38070a98fc8a640dd1e08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "3ae55e43168f72bdd914038c6d508c9c47e82fcc1b0495211903ecfd79460781", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "5727d7f199be616c0190b7fd7aaf652f27e7c94f59f6347fa72656bb2fb41736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9282c24e7c3a27236487e17032b21cb4582b03b489cd46e47786081a4d46a820", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_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, "84c22c602947414e1ef790b9368034849955464b2236ec470470485a4436fd2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "d54c37fb8cbef8815df3c42cef666802a0f07df71d2d02ca38cbda0a01d8e5fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "fbac6dba9c82c0050eaf751d83f5b60040d216397cfc091ab3497d13f42c2ec9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "64cf4490bd4d277da66fbf6cc45c3c5e3038a8fdcddcbf45701dcc179584214e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "65ba9f264865a08009e6d0a099874a05c131e97ea2ce74172ca075487488d86b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "e77ab142b4d7c3b5e32d64ffa5472f9eac105a45ac0a98a1912a077d9eab789d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_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, "f4b0ed58ddd01b21110b4f6fd83b230a330aeb4ab9ea0fbaaa7434e53ad1e753", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "7af7534cd497e228a7d7c67569fe6f25ef750e846f61a2cfacd5bbea92db7c35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "68f83448f5d4276f039455bf0828bd353ece56718ce4e9d3132ae84313ea7b33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "aa1565a4d366215622190e83f166c1863c14f7edb965648f6467673a2caac31a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_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, "43826ee99f5b6c9821db3e92dcca288776b15ca5ce2ffeb4ad7636fd958974ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "9106602c920f84876f1d12cf6bf60dae3573344a28c3386f351bd7487adaedb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b953e0055fc027369da87ed4efb8aa63cbc0eac7e934d08a027879dca7be80e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "5ad42f0d404c4dc208d67ce3edce95fbc5ce3c6df46c54a475b9ae71000ee344", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "ea146f9baa855992efe80c91bef77674d6c0450559a0f62381038fd791fb0131", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "692dae075023456618ccbd6ccccf279bfb58715a601efa013c4fed67ebf55acc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "98c7e6d268d1b024c0dff0421af0023d5ec7e618972f396f9752589359045592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec59954edde10f8ddd022a0a0224d7658704f278f957ccf0515b2bbf8cc3b720", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da57230148371f47d842b7e3b4edaa6a327d3e824de8b041411406b4b28e24e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f929e47eb654b0f885d44f09bdec7d92161d77c7c6d286d56c98c7af1f1d0be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "390886b366118c976579fad6c56f59032e89897736f816146ffe122c0e5bef70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c2cf7f1ae28036914f63a87309477f1cfba4156a1269594979e58072296058ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "461b8a53105aff7408bfb3f7720f9fa9a74b2a1722543313f823bc69da0ba917", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b528a00b24e5e8d1f75826423b983490c0ac9c09d7c0aefbe2f2c47a80a7289", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6fef420630f0b6044a20a18a25252058ef9832dc018896f9e9a3fde6b81ab49d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d68db4937144b1f782be0ad3f413343e942dbad04fdb6deb7f4b96c60bc41a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51fe50f72f04260d4aaeda72534d71f67b49245e9d954402ce7aae31a9da185b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7bf1745a2a405df3d1556705937fc138b58782b542d2d98ce135892cbef79b65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8cdec3f9299471a729d5ddc6ebeda7e1f50d079b98f16230a81953d94a81c4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "b5bad0f5ea2b3e065c488e2e2cc41f4ff15e0e647bb49ab7c9add697511c7934", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1fe2dd24e89a49b59d0e3d8cdcf9aa3115aa6d2acbdc3064612bc8acc47a97e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "a2b0b98bed13727936acbe4e107cd9e6aae04e5653198210e04efe7234ecb855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b2747ab9f64941f7ee868b1c116ca1f0344bf2af2022c283dd0c68230137fcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4666bb53b3cd96c050785b8d38c477b18636e24e8fd26a1c630d007ebee4da92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c73e849cd60791b21762541f4fa9fe606af79538d23cd5c5ba769cf6fdf71b53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da26799b1d07f633a3ac7e886118beaeeb322951085abe25943318c1384818bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4be0a6c843461a6a79a6c9abf77bdfd46fe49c164eba4a8ee887c8e723123ec9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "12b3e8cb1ee6289371cb3da9d03fae85ab182b73fdc014c15937c84b4cb384ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d0b7a976b20975051def8343c3c475010e7d84e1c01125bd9b651d66bae055c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "0e71733f880657ae8c8233cbaaeeade1d1f6a42c375318b90a1f763ae6c9e315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9cebbe355f283b544a1de1fb5a6b51a8393128d5425e709950232fd0650be5e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "53b94178ca58581f1284338afc027cd9b85831719aa1a762bb23d2574a8cd76f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "107ec7ba4cc6020e8e412eaf840eca8e3d443b5993b3706045be1d9e8a2c61a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b96724a68d956818bc0875ab13a2439ca4495778ad7e392d24eeabc83ce100b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "74bf4613aaa4733999c3771ebcba94708677a2ef2b659f63e7b0591cfeb99dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "1f1b9aa4324087eb455f86c430c387ae230bcbaee96feb609d1b8fb80ea8593a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "4c7d9d4fee7eb5d73e0754ec2f7958f78ac8c26d4b6cbade6032f6c139dc0261", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "09c86909150aad20526a125de27b74693a10b284a47f574c6363af9467b3564b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "1db936097f1d595ef346090c7bf8b1fe06bb54e77ee68c2cdcb68887e0ce8549", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "89f02d72cfa9b6656cd53c3d5188660c79f64ddc713c3da5636b26e33b288790", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "5a318b84b43f3d72e9fbb4e4bf1e0a60bff93b56f9caa728133b3536476320a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "ffa289611cfe10aa78e528e8a700c96e39c4a10b058fab9c58d81dab93ec75ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "17dddeb4c0cc9787e9be4e04fe3cb092910c31d1b415d2cb8f9aae4b06be4f46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f68110fe77bf5240d88705cd916a381c42787fb4f3174c5cc38c4ded052e3802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "034b6c4ba978be6864de6fe8cfe8c9d22e95fabdd502ed0f2eb8609e208467e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5aeb4455ce635e37db347a47e3d0fadd2bc682139979fc5097356271dfc8398c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "728ac5d4e37bed68324d7f28ffee06524057e3fbded9f61c9995d537557a9647", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b1ed45c53c55597671cec2e1965720c9a6bf68f6a4944d9840cc496aea7bf21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2c73c211d1d90a5068bda52ee9721ffc574c2597c45615ba6862eeb53aae1b9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d03dadc94fac351a683d3d5b63e5ef2bc4a3c9cca91479006c727d5772262448", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4db02676617f4b53f748d5619c6df32383120ad49815b7a1bf08db2957613ebd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dfdbf9c062f6b52ff6a2b3c835ba8ac0f52aaec5b07291a426112ce129de0eeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f27af3ddcc962c2ec449ca9d4907912f1e1bb023d9866a935905fa3eb3e30f57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "88252b00887ec1b28d946a2fadb635a226161a24a9a0746de3f076f94c533bbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a15acb0cee440b15da58bf3833e72c7be617e0c0e9d3613ff7d1058ca2de3a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e4ffca16fc45ea6ad48558b1ca0fc8e569255d192b04b892615f99ac6b00493d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "8e79fde163df40d85ec7c04ca755a6393233cbb031c9fbb9e909e9ea49382f29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "c246213b9aff5722c5c1e5a4598dfb617e04cffb4de30b126e17c9138820e33a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "51598b38f73b5a7ac60b9c749ab3d18d0cca6625fad6cc300282beb80511bd9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "03f30764067ab06d8709294b97884bae0be471cbf6b663f610e1bc27c0f20d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f4b7005eeb10b75e18b2dd4465210c2f7524774da2071b331781458666f01cae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "b310710bbc1f6ba96e71ecb5777d02d81b2539d5a736c8aedf70874e7edf84ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "23a6ec9f4922fba79783e8c99221376629c8b891a6047274b9c1746ea41e8227", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "0150b8fbebcb61684276778e3297a7f1931f79c936eacbd83d6fab0996abc1d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "1f11d5d0f81679d1d2897b3cf4fcda5a453d81be004e28feb91143b1b4c43055", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "d3f733430af92cb4f8135c5ead75b9522bce61c03df6b58fac78baef0135d8c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0a9543b69e46d7253d9a12cf71ae2b837c59672342bf2d91784f6a2ab9859c10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ba989707d02f7ca0e0271fc63cd96f6a05ff55bbe8117ef33afc24c7812e7f5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b61f9096a8bee55e5b1dd35e5e58dcde0ad321d85b9e649e3bd2444205ec56b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ab6609527852d2a6536652e501690688af66d505f131561b67a40a492b09552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7fda809e7ae9543ffa76d073fa8e94442c7e6910da4cf37c6ec74936d2410027", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e3afa78ceca1ef20ac4ce1e23aab3ccc022519af6e8e69d48f4c712f99b280f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d2d73ec86389d6802ef340684e7e397bb093eb95c4037e9a3c7293788dba888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7ecd7861d5e7c4af51b19ad5803ede39fe6dce29acfc7d82cc1a2fdf3ae1db9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0a38273820e94f20d96b70a4b55b1ba4d030a7639bb33ff79fcf6feda9ce2454", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a4420e0a232923fe7c21900ea150c90c52253e92c000db1d6db3df64a54f1d2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d0ea1de711f0ab927c32f887e31c913a6576ab7c47e8cf91203edfd963e1796", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "569a602038aab1c2f4156bfaadb187ee6afb99fad2c5a773ee930366357b4b48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "0b8a28a97f6b43a062c64a5edee7102dcdd165587158cb9eab2a665c8e18cb04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "df68b234cc972ab438fd737b2b84cc7743b51a49e8808bb9839559d20ad94f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "6d10905dcde99d30bcf6555c314d60f898f5513af762d0c247ec6fb50e8452dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "86567a2e15809cf41c0aae000eaac6bd6fa508f2a574e21bdd994243744d6e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "46eef19061fde21adbda539589802182f8ae60c0ff22abefa86ab2f80f2c8be4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "37d46b8701da8bbc6fb5f4b74afcc268b8f49c2e8f7f34996f54b3889653e13f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "124f06612e599ae8a3c2bf97cac963ab915f2f4697e76879a426c71ca010349e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "beba78b1fab42bf52d5e0e098694a20675ecd1fa05d7c7f8fae59ef93b5c99d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "8195a408b74c5afad1575e90130924a2cc3aa2474959a72ee1c970ed52c96b05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ecc6ade1d910780d08fea14a28a81672715b68741259df8db1eef381c177b1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "850fad38acde50b92d56a8c5e2bed302277325548d359424a0d0901b39e9e15c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9719dc4474019a6d29b2bde3a507beae6c5d20239774a42adb67c2097f4590ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b9c959b4e90761b94d4d6d01fb032c2456a717585e9814768d54836e63e6673", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4b6a71d591282c3ea2fe9ab8df08f332d221c0d7275020ecbb4401ab89d03f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5189aa5c2cebf48de2bf28424faeb036f14d8fb09873a970b952587c883841d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "98e561622047c194b49318b10f42af284355933287483a6829872c50e889888c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6205af4f11f1e61e979fd93331310ca25cafdf6d4462cb6b728ea0e149b19d33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a3d915465b13fd422ac04c9be2b9b5f37e41f82be0330aae491e88dd7507aedf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d91c9516b61be3b8eece860ae163274fb05774f47dcf19be2954224af679af6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db9b444a3f00de128e571bf63bfac5f6e1e739e531b6a445680a02a84bec23d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "098a5309a345a002d63a73b3139f0af536e97562e35056ec08a4a288584d960f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53eeca4a0690f56450d41b1520a095fdc6fab804ffe781ab71ce8b9e98f81084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "c355c2dd579c4de5b78cacaab94886ae4c7ddd48980f45ac5ad5eb22456e2311", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f549609fe73388a97b185475e3e3109f367c96ad38ba459a58989d3683a4d960", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "7f9dfcbf7388a448a56ce2f7d73c182d362075bf68c39c4cea6c06d37a6bcf69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "b404d40c9aaa9eecd0ed0c064253401abd90247873924be4b2cb360106bfde68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "37ec5390ba7d741e8ff6d06adca3221c59136a4793201c01df5c6102d9f07ef0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "673e11b2c7ceeff67ef48fca9f9daf7dda9fa506b281d578f266990b9f588eef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "f10f057d6563468b56d8dcd6a05b91e2a7382414be4785a28eb8c32ca001ae5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4a55b67833864ea154ccb88c170be7fa66c9b6869b445ac2f14ec6d422c24dda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "2baf5152b6abae9d97925825a634581977f40e4ced7d296b3199d08638e4233d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "67dd235fe464a740e1cecd6fb78994f4127da865d050b5fb624fd9238bea3b75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "90cbec31aa6693daf7de085d15559441c21599f5dc5bc252bc713013adcf9e74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "b4013e68b38ba9fbfa8bdd305de3b5eeb5de64cd1592165a5ad3f1637b13ccc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "65a701fc82035cd58d2a373eabfd7b9f546d0c16282e1d4df7a73d90f7f02464", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "dda1d3253b1089d87a7357c05311f22df5ed0f148c5993d6f6bcee3b6357f47c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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, "1debb1b4e56126255c8cd537ced56a2a24fbfda4ff5b0c22180e05964b85c290", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "91a84ace970ca2981c6f05a75abe1ffeb1ed6c3c2658000c1018bc7d2e5695a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "58df21868697d72effaa3f7e3c32e755aef925eff167db9176fbffee286676f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1b53f97cb025ba2df59d5747f68e5c4a4ea861f55e0a06d623e58f94b1f23a64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "7402f4df9fd7527c4ffa9c18db1dfac492202b57d94882a5b3c6a405b90b0f66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "79decfc8b981e9bb07c090d26cd2ea524f3891c66346e08ad0a18133220c750f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "f366ae0a61cfca650075e5d3be3c352e392c0b47fc013a068d2e5f75435c3146", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "a3410c3993f8e66a9c5246c2e9ef40b7465c3dabcf7f8639df8437e7d30047b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "3621957e96841a8a7c9dc2ba6631163485c6e8c704a6a86487dcf3685254448f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "7911ddc0afcb532869ba71d6f16cffed65842abbe8f521c8a1f21efaf84e9021", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "9e314a62bb4d4cc659b9afec06553404f56c1cc6f116ad890a6477601518cf8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "3b88ab05bd91f192cfecec46ccc3b7c750dcccedf6a6a9dcf65a40844e556f3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9152e377b5349530ff65684dde845f5582a8aa863bc1fde338f94dc8afa0c724", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "5f554dd338aee90ad648160bb80ea80294724505423e1754480f3d030a1c4d24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "c4fe9cc4e590f3fc5f8abcbf938c4aae6bfa75ffdd0f853d4328d5fc0230ab38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "ca7c70ba32a558d54b4b2e8955da2ef61cf6d28db9ce4daf955a289f9bf990f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "5e5286a8b303f1a5326ef161b708b9cd79712a0239b0b92a50cac90d8cdc3357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "4d661965fdc82d7e6f54e2022d0140f2e708c0aacd111fa45c9cd5d9653f17f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "56ac344bfb312888e56266ccd651447888c5f768e3522f7020ed62eb2c9a3563", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "3bbb81486f7f0b225274d2ed8591242578ad9df081cb9f1395824d6c624b44b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "41eeb43b5eff58891fda16a07e572a8efd7afe537c81530275c50809e8edb6d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "c2b45f067cf3c9b310ebfe578715e8bfbfc7120842ddf339ace4a7d44c4a4a7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "384c2cd1faacac3682bff6aff5444e9eafd5e0d6efdcd660503f14c5584facaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "fffce167fe3cdc0f219693fe2d972473d24eb87319a80c1059662ade7e72f9c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* 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_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "d2446c43ed41a9fc5b01e475130cbb49dee0832fd78b3b3b98db635898c3a99d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "21fb715ef9eb35aca9e607768f98b5a939be99b533c5243e941f4533efdb2e1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "b6a099b70c20ebd3d34d83906a4356d336ea5f004f999e276b8508c6ada1544d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "763f20986d373726190a77e6cb09a0ae1e5cfbce5b2b3fe5f4ca1d2fc3aaa8fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, 173720, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "4c9e9908ec198c6baf14ad8e7de2f78d566b5bf6e4df67da59408c0071674d5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7b92c53d5fc83eadc760f43faedf30e888654b6ea982cafbc072e536d0dda16a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 183952, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "6606690cd11a55fdc1bc2a1009f1133fb092e3b92970f2b363990ae7b289cc39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "ac980b73118f2f1210bc627ba249017e3572afc060443ecb99de960f696842eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e07d76647a7208d0e8bde1ada6983c1f4f212fc1aa05fd0175506eeb5503e5a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "33d9c744df987cd0a2e2975521192140148871864a4ab9d60af701e4e952591e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cbbc13cda909b10342e50d0f26ba7c368b066563aac7d36b6a0b7ddbd87df702", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa0de8cdcd31ee7e65197e86bb6b69af6902d785f41665725cb3d2ae63284d5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "95a7b77d33d7ecbb532795859ef4bb61cf10d8900465efae622a9c73d7e2211c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30cdc8165ac935fd4053f0034ec5ae3df007f10ca694100bae20db546967f45f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9db0deadb5a7cb8c4588564b80c5bf571ce484862c585d9ca116111cd00f69f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* 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, "262a8f3dbf8377265861620d16a70e0b3d0492585361defce30916cb6214a042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "e492dcbbea4e8be0c357f5ebf7403c601cef40f407ddcb782ef78f85ef454a5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* 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, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c68486a4a80931bffc8a30c7a54fbdc47959f6a7c8a0d8e46812fc0f3ad0567d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "525efb150f48935c27020dd22e040349ad11cf46d7cdbcb0ead41d9a6e245895", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f654512f0daf9ebf6a4bf1a2d1d4cc6f5e67d889329943bb46bac12fce6b9d8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b892d89566a5a8e75c28c01e1cdfa0668f2684fac9de4c9974607bf4b39b37fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "58041b94138698c0882f2aeeaae431aaedf29d896ca05595c94ad84f4bdfb09a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e67269081c7e69fb9bd1181ce5c7f98ef08035e2dbe653277c4a2f44c458723", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5deae678bef59953af5a0458a90e95f7300d4459573d807b237857d1c7eb4da4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6bb849b9358def98aafe784076e9e4a2cd98b9a3094fb57178206d54ecc1ba1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "44a4ed0dac1d1fab9d77e68a917d586c42536a5c628ba94b445a7cb215f6e972", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b06993f545060dc8e511fa1c2eb30af99174dabfececb03b64689346cda88088", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "1c9fa7d77987d62cdcfa0cde3dcbc8846bb9cb0220b1943797a1170da0e25be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "225a9bbcc9bcdf60e3118b795c0d0fd2417b4a5f9ec1cee8eab8ece306ca3e2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fa598fac0af77a5a581b6ca3eb6474147c44ba655a59924287a71e8dc582a047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "184e2b3d6eeb7a65b5d72aba417c4fdb69d629c3eb60f88bd99a4a8a7c3befdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "2fa9c56eb73c911950ff3a9f5ddbec605fde656baef3a08f2c088f7d17782bce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b082849c5b1dfbc5f93ccee4ce028792f0545f67f3f9224cb9f68014cc063d38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ef2663c348e0bd74cde661dd8d9cba90008928d82460653c4d1cd706ae86b9d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da5c6a61f2aaeaf3499b1a5c0396664a57aef55dbf2874886d537bd60d0e50bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "640caa4b36aff46c5e03bffaecb22c269bdb2486d93d48aae6f18ac766910270", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87ff5cc2bd45d7d7734cfc982aac22bb35790a0d87820769d034fd189263f029", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "721215444348be186bfcaad5b137e0af0d905c84a6a5ffa47db2f63f64940e88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "24830683b4091e4b6afd7d482c724c55a3412db90cdf3769644035847acc4cfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e6a51ad3c2542ea0825dce2c7aacc36aa6c5054f88e5684da677a09c0a8ca38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "66f1cf568c51198623d2ea2e8106be5cfb271be448ee1c4806789f8836a89cfe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "b76bc7a043019119f29a96c9cc95cbb67b11fbfdc617f6861fca2a203dc7203f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7b63c05d56ad2132ee533a9d9b8e749c9c6c7f2d36c9ec5f07e5963ec97bb4d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "1b4002bd382747cb3471bd51a1ed33185602afcffa4f49c710ece195b1a9aeab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5392a5d826549d66c3f665a694238c984799aa8b8bb131a01cc667f4ba25c933", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "4b0253f97621c60be2ecdafd358c9938c37a52818b27cc11c2904090837fef69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "df9ac2ea776ff65860a43e91484011897c5542b7187db43f2cb2fed436021e4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9f65615a4982140d5b8729153dd37f199cdcf25f5261c437f6e44675ec688d9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fd572e6362c9105dfeaa10389d60589ded2515344430bdbc92181f6cb99aa180", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "30dbd6487644e115cf5e85f5e60a69e39a0f60881f6bb51a53c3e60e4d00b823", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39b9558b5f175b74854de1b78de62065550db9a45062f927529b087faf6078ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "19b19dc64efb803cd82598af4c29508a6bdf32022a45708d6be268766e9bab52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7de5d107468c781ddf5f74abf49c287ed80e1edbde12893df9d2c9a1015f511", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3778daea3fb525aa266961c438d0f6a44ffbfc8f216fa389f80bd0d5c37b7f7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2470ea9f61110e24f561cd136d5f7341b9ad24e1e719d347cf9c377391241e81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "42ee52323a06c4da72f822abe5a0dad41a6854b64c679164f1a53312e0694588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d58c3615efc14198b0e8be572d1ca483b0a647a9a46670b63d614760e31e0a05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "34da561057bc00124a4fe303329464df99b47c9a0dbc2b527b8cb7e6cbe68ae7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d48327d3ad383b9d3b09d625aa9cd5abae36792d2cfe0080d4737e2c27575c3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5bb984808191038e6d6b076d70e924b85aa89b1e46171ca43b6ef4e5824eefd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d3fab5c94bea542c518d1cc4b52532661569c9fb85ac69b939ef241c3b06084a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "b154e83bf9dac37ad85cb003020d874b8f42f5b4d4b96d6376427e771525409d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "24a27e8114e88de0c978fd30459f49cffa4bd011d306571b458c848c94c135d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "558d65ca31827289c65edd6f06a969db56ba9bf9721be95d86efe7b364a3d7b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7947fae3c8b35651a5e484c1188d9be576973c1fd54172713cfbab6c817c9d67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2408d0157e5a7adb64f4a01c68b00f7b2a56ca9d4fd263b95287573d53e348b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a06056fccaa6ff09d14be7b9c27010402afa8f6d7c4accca26c310229fac3e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6e4ea1e86d57d10009f6dd09d13843c33bc5afb50d2eaf92fb0ce148b5940f86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76bd170c187fdc17a07e67555e88c1716e17afd65c9cce339314fdf64d5a0128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d12d1c64518aa7475563301d28494518a1345c56acfd6ff92674e8ef1248706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "81e3bca95c6bbd64e84303f63ce78e8cec9abcd473c82ca7871194f74f367804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bc8348555b2809a56feb9ae8ab1f5277167208c257df5b485ed92cec5eb8593a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad9683a6292c7c4e2b8242da00c128564f2c0077482e52947f47447d92294768", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3dd35876bc27baf4f6271eb5630412d1b0b532a964de0752058f650e3fb626f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07c95a96f5225e6d1d6fadd5ca6b5460cb9226f6127da30d5104c9637db36f85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dea759e0376a50f5e256394bc4f9c6b39a2c1c70797373ef7fe1daf00ebd4653", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "56cbeb13350ae21f74ab8d9d635098189f222280773f2ebfd0e3fc6575b5896b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9ef2ce7ba503a0cece15aa8cf0380da3c5f9eca0d274ad4184d9347593596358", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f7b769b8c1a429961718e0f3c271cf512cd2b40a3aedaf3433e6d84cce12721c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1a7245407c9ab10fc520e53cac3de767a098aa86be3c875f212a573224e242d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aca47c8b60d8527a1fad0b23532e35839c85df5faf828ee58b7bd5ae596536bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94f6083bf4b5e535d5dae796adbc6a2d315c74099cb1bb620a40861f99582a00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "09c89434d69e68c2c11b7627f591f65b1667b898daac02444573b93e6c0da981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d7562157e055478777c2b01a7d82b006b54a17a5d28571e17365425c00fa3585", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c62b804807c5886d9fc82bc119de61ca74835d97bc09d960500a4070c6c3e286", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39a590c1a77d8fb7f251526c917ad12687a77e1bbf2b789fce275f4e598d4dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b41da297745d8e67ea9c28a13c1d59e534c2586d317fe880af6afa3cee204f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b8e10e40213e798ffa969e2eb1150452c2537f1900408194760d5ce28d8e198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c610f685908b337b6d9f8b12c9ea5d7e5c882d429dff40af96b40967a8bd81c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d338dd75328e514c78b0efaa5daab85305e6abdfd6d4fa73e936178eed133379", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8ad2891845335e00cf01e5342bdb84b6c22381f6161255790278a5f221fdaead", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40d12d113edbd76de89e71dd724cb6fecfed226e870d0077c211b0cdba4d3e16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f828f3566fc3c5651001bd5f6ab9e96c748153824ebc419d9339aa1fe4e8b67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c7bbc2a364d58876d07344b4e856b8449762118e9dff767addadebcc81d6f10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "327259cac1fb1ca82c884eb43dabb5929511f9b1308ad2e73fd1aa496edb8108", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c028c84d8a8f3979794276197603555e83ff9aca02cc7318e036c112ea1c1d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6e80999a6c03b7d92fc4adfdd8183b2a4906c8166270e197876d258089339baa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8db986f11731d0283d470fe5c3d491c045da4bc76e9f05a2a5e6b867c9239149", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b6d6e4c89b4678f2a8b00ab6949a4295137d98601247342967e1a07ce7407082", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5765e90808738410a4001abbf6ca99ebc02f6a75dbe02af79e6b5a8a41372582", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ea2f8b343ea6e37233b61e6f7951257231e0adcd1c5caa507fdf1c386698a96d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4dd6375d12697a101b9b04ace06498c5769be96071d50138062f4ebcc21a795c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "50ae93913ca783cf3d1dc5d4359b8947fae332fcc4ebf2d8ce66562a9ba433cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7c4e4b17ae2843afe894b4714f00d4ba03079be37df6be1f3306e7aa88484cfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "772d1bba8d92d37cd7b447a81a03b2fa38c50d403d8ed1ddffc12589a10e350a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf35a03ed5a1da883c36a198515d8a24467b1705b06364cc4c6c183e2f154c1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7779afe1eba2f702a72ffec9dc0f7b68030d0d4629af5f7f4021c890c7be4cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21b0eca660842f1c87dccf3f1d082469fa7e6f60a58b936d15f29b4bd9397906", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b01ecaf1d912e2ae18ce2ba8aa0754f1a5eca50127004451b8f697c61faf8862", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca2b400e0ff3096ceee1f422d0d27b8253c66b5749125618b56bbb649b835ee4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "af76595769d34e1d0036ad4c0d7f8266eb2938d9ff02ae4da182f687c2d953e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "69556f0d5bd341091c8a51e9ac1764d6f86fb2917dffbf5a93fa457a777f047b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "286a43f9e5d9cf602a12986274714f12c136892f85d01f5cf3465fea42329047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "6e95dd0db45cd647ca43e7516137e825ca7375b456ef39cdd9b8c78340a2e72d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "c46fcdda36b1d6c66addc0ccd5c7a356d7c14d50b14a855cd7e91882d47ccabe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "938540bf4d48055af83bcb4a5cfdb49fb57c6246cc68e70f05506c6af3e7e3f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9007df2fc1f0b2b91a3ea65e658b6093475d19c066802e8461cc671d8f510674", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "c9013971a9a44868120b12a32f1d95dcb637cb52a567d86e92c91a074d69d2bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "73bfcd7dcb7a34d97cde680aaa7241d775dcfcaccd80767bb6820164edbcd03c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0241d4a51bc6c327144f650370d45c6b0265b42d316fcd6fd2d3651c7e89b774", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a0dabd7dff0cb72a523c0726da5c0a8ed181cacefa4191344309381fea0f1ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f490f492b91925f85dd3980fd51c8a9779fa721e4f30ee34bcc83528f27ae90e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45dfe3aa3d61b1fce25a23e939c4ce61031ac155ebc34f9700d58d82e8307edc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "517afae6edcf4a343fa4bf3eecc082c64af13b437bf918dda313bf872fd52296", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "270fae43f0e8ed99cfb68a482be2ae973e6b7ec3b3ede8ca81194b5865660df3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d6ad3d58b5b760e427623b2f1c0b2b53290a37a24eb6c6a3f299a71fd08e700a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "5965ce26d4ee7174c37efce24a51ed2a169e1fb16536208c43bff2add4c4b142", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "25d775a6800bb708e6c77c44c327019d16ff389645304d0c8ac3b2fdbb4e5b73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a08271a38f62fc8f89a7f72fb3d42650de7fab2b64e51da4484ababd34a948e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a1c94154839ba4ccbfcb8d75f5b89fa1c24ab7242e8340d87f8656d54bfaf314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cfdf19d86e830b31c57d5a2ac0d423cdfc420b7b2f2370a850b7f4cb7f6954bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e3cd88de5222105ab3872e980bea9d862cf82763b78f689c8a424a68242d8eb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "3a3bf637250d93118e7e4739abcacd0526d7ca2ba58b1fd6f4e8289c6ebf630a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "4b581d651fd1500ce22738936ddbc096dc8d72745a95024483885fd5536f02e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "0454020e6bc529e161004993dead2dbcf0c94ed3f91665c03b4dcb26afd394f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "5b8eadda99e633423654ce67933f547234da78f6ffc7db6a4822a168a9628cfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "0646411c8904087d5b9cc83f5ba1ede2831e8e43bd83080c96d6be72e1ecb889", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "7d2a83bf747343cd817211ee57a32ede43939d59d4b112b07bd84c8423e33269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "83fb1df90a80260bc392df2c9ae2d4b09e4786642dc27a6f6137a280d84bf8fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "72952897b7acb7ee84329b9abd52bdb0229a5b403cbb87bf074a1e13b6f4c53c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "abe87d959cea3f11904a8aeb228537704e1937de5a19cbba3893ebd5cda38274", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e74baf5a6676c433dccb83d4c40c751a3fd90903fcd1907b7a5bfecf1257edbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "2775a1a272a6ecd79a21b6ebdf5c4c50b7c060100f39ef994de49bde66246795", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "43d9e0778c580e4de4d8c2db90cb239bb7f681c30ba1676182f416ce842f124f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "7d31011d61405e530188bf97ddf10800555d8363a257270383e7901662f4c347", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "885a39f384326907cbc7131f92e1dcf1384ec0bb097b66e15dfc5970f2385471", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "1436460cdcaba31c89af4ceba6ad8e8688f65b7530f48de3af1326dd451c89f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "77d5a8c88d5067650fcf2f9aeb790ffc2f3fd26224534792405ea8b5111b2f1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "80c2a3f2fc6c6b120389f57059c098de42e1e3546a64d72377d2afe9449c6783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "b8e30c95921c610b5bf9882c9de8ff49cb9720a9b32032c8df6c44f9058f5914", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "bc5039e789fc52149ec2e1aa75e14a01657d2e8411b9c919e847c5e4d0362084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "76d012a567631abfd9f9366e9ee9c1f6dc94087b3494da9c47ecbf86c30d0bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "063c85b373a13dfdfcdfce1efae4479131180a05d6eb83eaaf31e8908a736d6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "92e6e10debfd3ac39b45afe1a40dc1ff5f01ea83e8453f17e2e53d876290153b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "4f73eff9f6d89c6415ad76e3e7b22be547b269592f92c80b14984cb7062ac20b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "78cf6df18a3a81008a16ad22b0bbb32714bbe602b76533950f43db0a72105075", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "b30bf380f7adbb851a8f1d620fd4675b442039a4699b22f6d2369ec13b015d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "de72f4761adff3ff39359081dc8499d3aa6bc603d8c6e19965b9174388b03f18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "c8c064ac4de2478c5e4092c7e0ea0182347f78fe75f12cce55e878deaf65bcd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "0eb8cfc5e4e77284221a2c9c708b407b72e1068bcf8412d840ac078d772fc630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "5eae3284df805e1ea141bdbf258cceacf6bf672d8266b4d7906cf9c91545a6c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a18f03dc25800ecc92cb18691eb4adb869b2773426365951d38fa637a97dcb44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "eb41a7e307a29385de1832e9fdf60ca9f1efe93f63383359d92cf32e7891093d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "7a3810f3f6f1e217479eca7409c451544a4a88942c3f1bcb4c7cbe17cb583c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ec42bbda584d8663db4a395b16dfc46dfd9e5acc9832c73ef48b8c3decc1c8e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "3d144c0e648d37f151e355f136520e4755da60bd9976ad7ce8e4171b118a5ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e837d3e5d08d2014a6c5529305903722477dd08bf14949d661de624f201fc666", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "362decc0b9a5fed60beaa2ef7c7bf84cb31497509d19b41876ffec6b2d071b4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "53c7401635e6e76a9ce59d1a9b6873a8ecb2fa1bc58a7251dbbe2b4ecd9d92d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "51922b47af73ff4e749facc895a388b4f1e2dc4541389af3a845e0c0d8665006", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "2a63593561a2043f17cac5d85df18ee83f767505b9ebc6e1c840d45c0728832d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "324c33950618f35b4421e0c3a37381cc7fe13d6092b645286ad5d40adf2ca90d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "530afebe358a9b70202ebc3d047e6c1046bb02ce573faa4bd13c2ec5fede8e4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "8ec109ed5460b8e0b309ca51f1d39395f5054a3796d469f7db4a13144b021567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "eb8d56406d96596d843b5a5cb58e4de755767fd051cdb2f425081ea85daa0467", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a1144e2edcabd4ad4533929a7c4bd50df9000b5d7f686befeb18f36a8dc32d70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "52e0a9628c55889107295b1080ddcecaf02c3f1410009d24ea9e1262cf7a7fcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "027a1cb493b5e6e8283d81a9b3ba9a2ead078254396335ccfaef89d936f5c484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a8ceee83ebf60109ca7de372d70d0eef8ae472bb9178b0c588b42d40c07dc4a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "68a36b817dbf660f76e4f069f2d4d5def714f38dd005eff86ccb16a37585a0df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "090f92df14e8d9dcb1b2e9febe27176016061f572ad83d6188edc6897160915f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "8fb8d65e047ba93b384129fd624b552a81716ecffc2c75b6821d77191591eab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "c446b4f273083a954e317865bdda54e9c2131dfd8285b3292ba9fe66cf20e428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "5c66ebdea01109ae8579b719c7a23022a307b26d70f978226d3754c51d411c63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "2d85e2205b6724bb05544d31b5d9dd2b8c1d4190402362f4ec286796c5e79f6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "bae8cdbb054a2b0752c3edcee43ba363ca1b4acc2a63dca838d855e637b24a00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "713f96a3acb0e519c27944c351738b10bf6930e334d939dea565196f9cf50f4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "fe08156ba07f00350b89a698bcd9b6aa3b35ae5e45534fd76cbb80af5aca8073", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "8ddbeea5e78ccf3deb1d1228ec112fcd1efe7bfe0bfc2a838e93c1f1002da7db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "06d208e6a215dddecef67e4c792b802a1598eda74aed90af771a262b9be39f37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "cf3cb5da757e6730a6e761686ed544dd81f569280a90a8b73c4f59e66bb040e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "40bae4ec7fd8dee1849ebf85dcf93b94c15ae6484c60e7cad922dfeee8509a3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "38f21dee2d6af54a3d84332d5840b919cffaa0292d2eec73d8ed868b3a08c90a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e76416f4cc02fe863533158653cb2736ec2ada78a7dd5067064a4d092f4dbf09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "4981b125b2ad7f9046c81543254ffef53c4a1b913a37fa475b4ded7b8645610d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "8ef92733915c865b575667b3e97acb1912fb4e5b0824bd59e8ceb19ba1ab72e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "30e3b01fdda981472ffa8c50fac9d620f4006ce64b214b74eaf3e80251455dfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ddb9d96d08af05a79f45fc7ff5ae729e9be46cc5b3a81a7de94de0d240a27ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ff839791995068ba4bdbd896c255ea8b8731f0ddac1bf49cfef73f7046104b97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e79b9377191dc04d78e5235869e9c8235405ceab7faee78f32f3ae7af87a4e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e0dcdc9b262e7cf773f599adbb24804bb05c7b8b3eba6af95df3cefe3a06950b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e2d8c0a1f07a3f03c502852a7e1a355ef59705609f6700a69c39244c964b4169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a23e5bf0273a6da746f8c63c5aa15e36616c30d5c55bb986a20d25c74e56bf8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "3d29b958c1d574ec712b3918b95f8c50599d82d79b626b1f0e50b869f1d8bbb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "c3b95a5c739410d28e0301391ed0e161426fe3b66170f8deedeecdf0ba15e8e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "52b84d0d66d7c97972e75b3d5b0146f1cc3b7c35d8c713dd869e98716e3b52ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "f98337e1d291cf942e0f877d7ce35a9531c90b13f357c1b7c6a84adb5609d8ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "a0980f96d699267ccd2f55125326aa38503cba0d09ce20a8561e08a9c2c1e52c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b4afd7bf67c0be7f2b52ebd13e3a4c9aec629d23e70a5831d90d78817da5a184", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "bf33e51dc50432cfe0a8e933272dd23195908c7088af48e49268129db1eb7577", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "602ce00ac98110d36c70206dc8c8af29e8d0deb1b18a797a8c25179e290475db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "122147b46104a897eee2c2ccc3f722177035165fac4f16258d7f449010af44e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "427d420815971c625f22fdbebecea58d207a2e9c0eee9294cc7243daa339bd24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ab8f895177eec3169f8731e1c8df31a7f30821b648bb3b45ee0a4ecd76219236", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "db7f54d26227954f786db4601a686eb963dad83cf3807f9c7ab7b3e0a2245384", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "4867b563ad4c68e8dd12e9fe96aee320e82cc7bc0e6d272884cb36d115846f5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e3ca77e041064b674d9c4d7f8164bd4d88bf669d346139097331c66976766449", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "b4c44b4e0b3e382a45462d9171ed85af41d96fff41951df81137d90047efb4b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "34cf739dab5ee95f76d51fac52cb3127742973f64661abc9e87cd5b17586c072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "27b96a08d6b1c1778d5b550412ba74f00faece759a99b44a83a61f76786bab94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "219dd4f074935d8a0c58725bde9e4ce4f921e93f5928e4f9680c1dc064ee9ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "9dd8f2ed1ea366959641342459949ff5e86afe7f7a6aff7898d1d39e37cf5966", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "fa0f78c45d05487683ff8026e1e47113c3bfbb16dc866c87c7c9436d68a7b6c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "27cd41ecc30d9dc45662a2157beabaffa9d3e0385dd2a86888318a499640f044", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "9246cfeb3b202864479fe30c7ac039ed00cdd23218d8c06c0989d30c0ec7c73d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "b60f9133e43b965dde5f93027c343423a9355cf7b3db6a29c31bd96fc2134fd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ca6b9a0c57756982037539c0dfc6f679db77148cfb0abc97c701f3e8779e7da9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "84bf143ea81d9a8468e62bfc836292c94d621a9782c3d092fa70a1fe3652eab6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "02faa420166cc5ec43d67fdb069b4a2c49638275dc329f6e040b63c542cf3684", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ba96e0e9455c4a996414606c8ba00a1e75e510678f9751081f36883da471f3b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "fdccab17b5b443cf1f0348a627a64655fca803529595a6dc67275119ad15d12f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "5a28537f96a6a1104a673d6888af40acc90277c78d40f0c1b8b1aeb1e2e57490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "f9c4d62ad0fc2fa1f4733188b3b331f2286b6d15b15838c85c3557442211f7e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "3f82a9af5de5702ae309027f273df511b7398b82c0cecf4c56e305247c3f7291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "94aa8a17260265f6f35b2fbefba20c7ab0c8d9f021449c4ec27de424b16c677c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "be245bfc56c3343179031d2f1b3a46988ebc7281d08e160897426822ab17b421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9f23e069e9d817887f0a0ed166ffe37f790f00397755080540ca7ab00fc2881d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "c86201f4447693d48623cf6260010d26e166cd070ea33360721b847c612b0db2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ab479cfb5d26591e6e45504ff6908fa77e6a3230d0b5566e1fb14ea17a2bf49b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d8a640f84a26df7160efb017480ff56cf0edc51524403494decc98fef015577b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "21185805fee814a085d6f32a11db524d5064cb25dd1433621d24e3dd1e706467", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "f5f1c2bcb6a39d8fe6f5813da1909203215e035002e67ecc9394ab5fd46c9d24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "f8946dead018ce40a4436b51c9df4f27a60a1cd5f6448b5f77b5e57b9006253a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d505af41c33e361698ec2e695ec55bc6ab91e21629ec88017b9315199f37c428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c7f5456b633c531677b8e20b990a987f45da7b2200b3afe5e38b9ea4825714c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "710d83cee801035b5be5824abed93317838761cc9cab77f5c56c5749cb6b3693", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "0ec5c34255cf3fe792721a46cf2fbc38c1f949cf2d2bda1fd5b315eeddf7e009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "d0eb48003d30037a778de417fb5a23a65a242c038f8b259a8bdac420412197fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ee660fdb3d7039aa5b6a378b4552fc9e7d70a96b1f539bf34102473c9afea7bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "288db34b08e2914bf15a6260113ca620c3b25d53c8be013bcc60f1815081b998", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "2ae488f7ccd9eddbc623e752424864a27be1f16c401cc84bb65510c54f051595", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "4dd9f8c401715907a83cfd7fe6056c1660a508a09b35c12e28726f3c90bf7bf5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a1a18b89c7b14652ce9ffd3f0ef16a4f6d50ccd2f8acb0c68db04284e0631e68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "730f7d68d5f8360a3e991b72fb800d3f57f77b164cfafb428b57c775d2d279fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18047b0205b27a457ade7877d0a427dcd1cef58c54be00581eeb8a7a37d5d6f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e64b9fcc871e47ae302633a15bd1cee07516393a1e2de1d80801a07e0bd7e3c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "40ea14f9433760e52a6ff4f34a5bf21e2d7ac0f7ee47cafcf3a89f54b972500a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c068597c8d9e5f93371537e9f54ab579d323f22a83ede285222399d84ad637cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "311289323bc5f6bfc742f26f426a3e33b6373afd568b3172ee1e5eba893512fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "608a0f9ba8824d12b775e7f00e605bbaa451d894468b3a55401ff44f79b59548", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43bdc765498004c39d4dd2f5be066f162076f17fcd91600b658ed6e8470f6a95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "17c24ecd2c78fe6f9eec6b61fb526488062d0c882ea045d1ae00f2ce1ac073dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "238f4342d2a31108e5361a57ece53569488c2e86b0ab1da80f18d0cfb167d446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4cd43b1961ab73c3590ad253fb851f1d064b248bb39721fb038a422d837bdc3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f5cd4e72a88ec644839dc79dc6d4f83edd0ccc9a4cffe33bce4b574a80c6c89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7ac5d3292e0b685611158291b9e788256332cfbbf387f394552184eb969fcb47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e708308c16d4cd2903152d84ec19ea2625b3977f882d953d54a1178e3d52f4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4344ce40c9fd62f9b84d61f0b94f87a3ba269b76a1419db28dc84ed9a77191de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42231a3295d76997c6040b9ee96e9631ddabf6f302d1fb8bc3fbd391e29bd494", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a1371672672fe13909e99241d36eede27c000fd3aeb4f191a0eb2b8926e55982", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "061309aaa9551cd255460fc9ac45e86c916ad54f200eaa8496ef1fc2488d6d41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "946f2f757b61e26e5e38b97892ef9f5d197f3d926762ec2b3366bff5f1fa6f9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ba89201bda7db6f5bc12c230cd2d1e8901ad68f7ea24f887285cedf8f97c4cd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a590f42977f40b7118d46293b79fecad6dc0afe0505b4e2bdb43989713f0863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fc90fff76879034a33bd4d5cddace54dec2712af4b9e77061b5559dfffd62f03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "286a9165e8ae93ca64b341540fe23eea8a36ca4944282ed971d003bf9bc83567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "764d009bb8baf19c1aa7044ebc2968dfbd56d005147ae78144edef1acb5942ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "37f8c52fa943c4cfa5ce84ec9661ef78e07bab2314652e0be75448701922301d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "66dd0b4199ecb99fc81253681ce9d0ed5f19cd11855772a93ba67a47d9fc559a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d8ededd985253232452690147737ef2c61718a37207a35d9b0ba59dbcc4075a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b7988d4b57bf502a801a01a7d592dfd1bd8ca0605200ce441d4aa96a6603fa0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "619252f1664a2b86b781af902bfd5c04e8944055b0337e6473e40f38bbf895c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "58d18a57242eb2db4e8a5e98082ccc1dd483f05172d9d2eac72c06f078ecaaa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bb7832e6e6c484de04405273e07d31ecae06370bd55a9fdff2dc409b09d3a8e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f27ef03c79df9223f3d20b560bebbd0417596c6df982824155bbdf028732f19f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bc2fa3f3f66903b5c5f30adec57122343e21f52ddb60550a391028e0d57682b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa70544cf1dac0eff7948ca5f3dea2e3dcf50a764ef58dae504254e1a472ff4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da3e8e15b3343d8c0b440abfc6743bb2d5c3bda7ad2c963ca56bf70b7ec7e495", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11bfb77e5c05c978d8e4ed0105fbb80b8d4af54783a1d584d017708d57d244fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d270669656beefe7c712d515c2fe7c4e74952e408ba3265ecf235983170b71aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "44588862792de29720eb29a97abb12464c53942610c75757c1ac346e517500d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8a0bd45f933acd1d009492bf57cc3a490773be5184c7bc8466b015c89cc917ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2cbf7038df0d9ee4d55222230c9cf7915f1d78026cd7ffd0ae699aad155cfed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6842311f1133c03ea5d5285d672dec7667873043c517c6efd2607029061bcf8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94d538fbddcf02dd2107340d7ecdb545a08a5dd00a6c1d33ba4f00729c6c6ca0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07724b0e0de0d10cbd841f7d61b3199489cf116cc03c9852002b9ebbb250e054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "32352568b49465280adb6e78158c0bc619d6de2f9100874d46e2f9d22910d8bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10ea7a28efe9efba9c3f5e8803eb3c7775e14827f851a9122a1726e04ba2536f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cbb3c4ee2570332ec7ab970504ef2e624dac2cd28238fe9ae039f0a73d72b4b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c9cb8e51cbd5a48a3868f49334bb0ae0e7fa4cd62aaec0f941d394783252946", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a4ca87ec9ef4d01939b1ba5d0861785038dedccd0d91fc3bd8419cc5f4ee8163", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe976af32d6704e96b0a4bc263049c06a32a48b97661b5dfbc70e36b15ec1e4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "37cf2ffeb12d8841a3f588be3102794f6f2f01564b13eb119cb6fc189f4728fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "779b0dea56f8228a8e761105acd409e9a6528afc667d1db3a291f3a976314d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8bda1c665d9baac149a480998bea065f21d1ee8af3970470fd3ac067e12653a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "de5925c7c9003587c0903641f75cb8e2160db65a18e580f22c0272e7fa7c3d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "760d4a6ddabbe405c0fb67c598dbf135079ae1fc59b2a31001fee6e996eec49c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5fe1beab81807db9c2f69b8274cf9603ce622ad7dcec6d32017abfad2a062e48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0788615e07aefdaaf4dde90b39408c8d52ff511289cb669f8afc070fd9186997", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b2bd906a616c5e2e2e0af760be7aaa5ee5da149775f003441ab7d1629f73838", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76f1426a4827cb326831fd265cbc5c936157589f0f1a8fda58a4068f493b3863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "86f3e89bd329ed4e109f61c2642b75792279babd84b521aff7b5836b77c290ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5d0b82b6461356abfe3d8682b46da9e6736f36d940ac75223da9329990f38b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3769128fb24aa8660fcc4b2464f7c4cc08aa558628c7fdc45523b528a5cc2b59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "59ae97afc435715104acabaac84f3d64616cf61a0717b0bdcd8830ec64f3c6ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8bd201e1f8d48e040e101db21335cd429b340aee2ff1fc4d3305af22cdd8833c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "957090869076e7db1ba5e609961f20726ca8967d2108e7a7bc085fdab1ed5117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca0a68413d818bfd9742bb7e97c7aeebeaa2d537cb531ae695bf8c10c5931f7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "869a050a79201c47303afad821215117c1ed4f9c71e7930f3ffea9b5c602911d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a6ebe22fc389c5dc36d0649a21e96b1b8f2534e49502fb5be74a22dfe8775d0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0cbbebe441e4b4dd7f27ccf6911b6643482bbf3c0d723d7bcd9f5b23e5204a9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "914dc11a723000204272e2364fa1a599effa5ae4e063229f6e2ec8cd577c6627", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "358c4cc685d754dd98a5851352ece83f6450c8bc0489ecd442bd31ecb8c83be8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0064c2541c742f751f90f7e252e05704f4df7c9382d596449600b353e913b661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1b3e161ac6e65b2ada6aca8b7ce6b7336f0261d900bc6ff4c2b63eb28bc15eb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "17902f7153d0ade499e7aecbd7c515d58e42d801cc8b5dedec5771346b690235", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dcf1c9d17948e58d91a50e6895170c06be0759409cc02dfff32506aed9da736e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c5c04e146248c5b3cb39e4c6cef770f7a3bed257830c13ac779dad0f82a6554c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fab7ff324ecfb2d36e2952dc00c9c28a397d689f2799e52dddc8040487f356ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a707a9bc6efc47bd6b86b7e23b92aeb53a00c2c45ae5ad8f241f3c1a1218a97a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6405a8c8597a80cfc62a1fd0e74bf194e7b4ee0297065f3edbfc16661f8bf939", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3d208337af83371e3b28c43aa40771379253ea3a683078d55c77457d79e7e26b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dd51c428ae26a35d68580ea8966256236d5b55ea5bddfa21eb2b29d723c4b12b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3681e8ddc32244a984253e91101568d930ece7d48bf8023be8fe714cc745ac2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d40ca5e457defc82fedec8395b5bae46e406196ec3fc20c6333ec7bd82fca32f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "986329c641ec262afe94b53b29aa0a78dd5a1456f42eba3c87d8a74df7772a96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e11d17db1c319035d0434f7e0e7b94bae1db40d97ac5cdbe3062bc5849a8f2ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8649a79d0f70a2205c91927eff38ae7caefc9f16035afb568e3cdf0966c35c53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "190473fbb3c8cf6a4de1cbef1bae0565544b15e17084f102f8146b2521e4035b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "98e3761a1fb95b3ddbaebbc40eccbc4b407530c5f24f061ead0313dcf41ce5e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36bdd1add3d0ab658c1979d161ce163509648019817f10b62313bec75cf505a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2d9e525c2a8bbc028e14ad30d13596b3f368c47a484b1a6c1df7efbd1dc7ae69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3b724e99192c57f22839653a387939d00b9b791fab371186a09caed6ac750e9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7dcdf2725849a1bcba36564b8a8c219ca777746146eadc64b583e7b86350755a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0f14937903afe04d1f833ae44804d64aa7a1b4230ad2707ccbfc81526642f422", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "475b2edea7c2e057582066edafea37a913fb0fa02fa8f8441abcfad8a64f508e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7f146b5365986f35acc0b6468fb9b4143d62ac2af8b5690c484f645b59837a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f8369e4a59867e9d1499ccef19e9b71de4f3e17c19f4ae94e633f3ec818eed3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1c0aa62aa96a5b099da8575cfac47e2e41e34d48c6417b34fcc73d74de9ada91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3864033547fd6c71e4f07f7351e78f13868f11a0a82289c9ab6020987f29b6cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0ffe845f0249e34a6e917eac1e14b7f763c318f58325ded911ae9f35b4a18429", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f3b4435fc8f89ea2f4f8296a02df31c49ad2d89a739f3d9ee4a7be4558889a12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "23ff88f295f8cf3f353c680992e4ce53bf49cd37ad2603ce078c2a0b07f59de2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d1824864a4a6a82246a8a919617c50a8fcfef27a19707812d28830d0167ed6c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "05d6af55078694599884f7ab93db5d519f11ca0ff52d22ea83c32205bdf4d566", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42ae4602a915639e588becaf16bd36c0920dd82f0ae67da97e48425538104529", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cd6bca5126cc49b1afd1427ccb83c917d6b46e95ce0221537ffd26c5d7deda85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d2ddcf7b5e7adbe09f05686c052d682ae050dad74cb99357b78ac04120736072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "595babfad2a8aafe5226d8238794cafa108ad484d046f54e15e73913b8b112b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b342820cf7a58d10b2b739e1dde5d8b9e4eaefcfe0a7e2924818ef9c3b3174e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "738278c3a4642ab95965e29bc4ae86caae1369a66e5caac74395e549b663adfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "00b720c5ef8bcaf55f8f8d297c51aa7b5ef472899d3c0832cd98ef34c9ae411b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2d5ae2d371218e8cdda20c48a2fa721aef85c77987e507364fd4b9ce8f230167", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7fc41b5088ac355b83f9e2054e25f7c62b77c2f813ec9be0ce9c9bd799417729", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5e739569fec79c14854835745252626be6dd18815bbca0042ee3d05475c57739", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e8a3947ed80a89b26a6a5ab91efde5b59d0c4ef9781c090b43fd729af76f374f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ff99adaee4c5553e8a1ad56f54a89deff17d8ee32b864386c4b214df21605228", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1be2a607ee016af28f74c46525fb43f9fdfd5b0c62f24f252ea6243fe2cd9c2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "2132d974b7539789af6bbabeb1f5dae47c109740bcceab009a14acf9517e683d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* 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, "debb38afccac896e77909230ed82865bf924af771b2197b27c7fa5cdf2af6bce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "81a6373ab87199163eed49e37c2f201828b2729c1d1369794bf3ecaaccf71640", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b653ba6ebee3b48fecfb7b545921c6cf45f07d0fcfa22b30bedffb0d8bf6992a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "36e309ff392c73ebe6ff5aeee57a7277a95102491f150fb5dac701c79bd668c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "de468c46e471a38a327b528ac4b917d7f2c1500e703fab392a7b4648bf167979", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "51edd70c598f7747c86ac517f485b078eaf54ba5dec1e9ac0daec7fc3cec0ce9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e71eff968633bf3ff06c333dcb0c63a800bcb4fc8a3dd3b1ecab3b9101485586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6cb7bc85a6e1dbe24dd9b8df4f1d3ce18fb18cb7dc65682219862d10c15ef31d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6de00690b7090d398d3d22109610db84da25b3074a4489285495aa7b6bfdf769", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6cafa0f956737b6c8ced9e7cde5b81c6101ac1b2234d76cc89ef89b268e176e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6dba16ca32985a1ea9aed86c780e7a8b4c62b407b4ce8d0727bba2ec6ed2a599", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0b6a1de93895fcbf9dd5f7ff09e503ab188ad24574b6e7d91a455dc6d862029b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4a6e25df604c2935195938b170ed488fc6186af43071abdc2facc18e0b9ef4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "871dc94131b14bd8f56d931557b6904979dc25da377d156c5fa2d7f42f4a22c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c4c88f61c2e90ce8ea367b438aecc4568a4b52945f1b7cd70ab3f0856e64a00f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "206ecc3ce182c5debbedd6e790f46e0e1fb512a11ec4737f4ea65f6f819bb809", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c1b838fb7c49980c3594042b548e073aeed9013d9ca17f8d555ac1e675fb325e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76fa597c0c679d837b2a2974dba8fcc771c479c8c85dc2b43a4f4bd3a445a2ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "af90a7e4a12eee6e11188b958f5328cdc7edc6852f42c557453177bceff0ed00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a96640f160a1284cdd4aaeabb775b9d0b8b45890a8c3141bd567ae8fc6d76428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "28d220b2937c8ec5437f19cf2fbc6ea7977f4335d563b09318186a927cc47df4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cec22376697379da25f7c53f05af0353fb7e76433ca779fd3d18a2c3c3e615fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a71dd896dc5cf834082acfba2f767d47b81df4815588e427939dc0ee7ad74709", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d8817df348440c194e3fefcdcf8f0990a1d4f54f7f8ea1012597b269c3584168", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7bab18377525aabca7234b5f73fb385e7d643117c885cef162ffa66617f303c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a253d1d8db5c401e1b8d5fcb36fb511ec5b2b91d02fc74cc9562272a09c9f64d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9142595f4b13781027a6d406883853f17394b14945bd7cd247d8017945368bc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "24bff8f8236a8366de4b5a53d614783855a9536c25650d8102bb4dc50c248f54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9fa565e682961f70f00ed87ccfdc1058cd4c4e91786d6e2a6a94751736f5e45d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "072fb926e418985f09eb82435911dd355f5aefe1393b01b9c0365e9212d578dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "358143e593bdcf7bfdbe328c225b788cc083bf70e002f3ba158c5df30521a5d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "90123a74cf88f58fb7f21d35450a2fb35e736f20868263b9c9c8dbe76676b150", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "f8090609d5420830c8c6b1ca5da928093fb0b1ff03a34d2489c2fb6ba79e5cb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3b2253573917e4c9a461687cb31353fc2397dbe7dcffcac5c9522f5a5fb19d93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "bf412a35f9a7f55f808cab6df0a0d8745532cdbeec1a1dc132c34f0462caabe5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "632bd21e68086dffc91c0b74ade46be8b5046e392715c7837182a47700fa35ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0090c955bab81f70e16953c5900d76ab3ebcaad6da27c8bcccf92e2dc436b943", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "bea38e6ffcd6ba04fc06876bc408dd4c5cf357402ca03371df382ca118d46d4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "6fdae8837cd73d03c76360fc6ee864c671a27ae1bccad6a8743ec846db25077c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "14c926c613b76a45c2d3fbd92daa9542cfae8ba721d6ca05e2fcf150406273cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7368443ce896928db9daf7717ca8cc3c5a4cf4df88aa3fa35563bc599798d8fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "45025c702052b897693cfa08091425e1a921316dab089c745f3afe7c4250c7a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8ae2cf2ddcee7f5f5d232aee8f1d599310d6550cf9b8a14b639e368a793f3612", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f299e93d31e8dc4eccb0aec19e87a69e2534b4b8866b7f43d4fc64bd2fef0de6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "80ea5531e5b30c22f45d22b08a7f4e715fc1c0bd2814f5a9c4e82d044eb0e89c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ffb7aa57afec237cd2b7ca9f6b98c5ae88380077004c9f344128d2992bbb90b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a078dddeed773bb751828226311f325563ef9aa4e672694c5d816f150250a03e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "bce1442a2f434e5b3bacf0eefec852f7ffa4917310831b7b8b7ad5c67bfe9094", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a8312a1c24bf99a09662b1f66f6b2ad20264055c1f5202fde77a0f27f4c6d04e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7e752ac7d0443c4992f62fdafabc4657d09d75bf27990aaa9e58ed2ebe653ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "9fc3a45836c00ad05ba98715ca8f749a27813ba698fcf45206378e40a861dc38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "2fdd893227662c60775e39103385053c7bf3aae9df94b2f456b2c32656ad6d6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "8983ec04cadb8051856c1673764d0d6baae2cedca797d66ce3d5a956d91d71ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "a854986a03dfdf1176e1a759a22c8e92e6c24fefab70d03f6fc97fdc3a0b8041", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "ddaf3c1aa82effda960aa78dd081ee642923f522ba356f6ef6dd89ccb00725f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "9e3b3c4b09a9a930a73f467d2a329e2f138d7a1b4156cde70822a414d512e0c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "704a4849e6d59aea9b62df2d55e9cc9fb0687443d3adcb29f8c539d6200233d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "34ba9a9259f0783ea33fb5c07ac2dc368a32eb9f725ad97e630d9a843f73a725", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -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, "e25da26a554958cbeb5ed770639bd548ebccf4ba7c0e7e9e0003992653e2d0f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "e706a621ffbbff74a751215ffe282d7c5f2e645b2287d0fdd5155335530a7c3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "3f8f44eb0eb6154aa43b1aaa8aa8de70cdcaaa52f376ef0b676754c016f631ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "dbf76ec493ca17b9ee1590dba9b7446320e3a36cb4b49cce58b77b32f767bc50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "6e41a5282f63647e1d31f973a5a91e17376326ce2bdf410ecddd302e368e566e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e71d933ed35ff7f10cc51eb30af607ff98a2bbab32dec824ef75a56b4efba267", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "445f37077df6a7984b7116ea077b5591a1b2c4acd9d7c150bf5cb0630bc444af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "76067a126ea5adf0d528131a54eff25a2b949d6c2b11386a1b79ce5cb60d2da8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "342cf9452eae2522b79057e3cd8f46c84259f4f9a4e4118bc6b527b7d0339f81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "709c59a519907c54c5f7bb3e1de1d122b21df16ecd1f9ddadc177214fb981642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "615594f9ba884d6679a72e8afbd0116bfd77f36e5bfac9ede7744d8528dca1fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "a72adb41ce688b6837d263e5ac0eaebaabf9b0fbd0a23dc6c85528a0d7d34326", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "dbde8614a3069b1ca354f72c7e9bbe562571bd77efe85798b4e5ba30ab527c56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "288d71b0b3a2f1a367f088ae69800efe6b287ee2422c6e38019c8c0f2bc824f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "c47ca82ec094a89fa09e73a31bba18febf73dab52d495d6d41e77ed941172e8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "71877e68a142f143cb97d4ccbf4ae174ce592d69eb570dc5dbac1f3ee0418f96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "c860b18eb99e7eaec402f081fd2ea3e61cadd2fd7da3ccbc68e502a39c05786e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "3b53ac1aa3b48bb619fc31c00a09c78ced8d09efeb7e3e67deb1aa0927e9db8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "cedcabd66e1b2265bb8dffdb8f3d9d917d467f0b77036aa68e1863b225ad4179", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8113e87afd97b11ed8f349cad4c58b55d7b2f22ddd431b1fd71516fb5764cb63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "9316c8a9f75a0581f7722f667e00d04a1fd3cff67eac08242f52b570363c1fef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "a25a0d631ae71b12727c82ced155f6c4c50d39d926d25a596bf26cb6b2eaf7ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "22cdb1ce09be99d7767f2a10e95b8937f1851f1760d4cb9f96de6714de6eff42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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, "bef976aac4da41ae9c5bda11de9b5eccf314e19c15a46c9f88fe6384fe7761c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7be055ab1d0c703fee5a6d9329d001c4b885aa5a5ccfbfd7a6372cbaf20d26c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39c077d1b0d4edd2a86e738c89fa83e4ef1003656f86ac0d7345307287afaaca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5295edb54374bfab5762647f1a0c754bfcb6d1d22ec6c1ac8ce80c3b1f7fac7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b1a0e861085a234db84711393e274f39f4bf5f7d2ff6aaf24c38394aaf9e26bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bb2b26e537bd69729e625da0e33c39dbf6c6a3d00b34c55c4640f3c57a27394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10872e486aeab56d2be9b15c5cd1e6e158c0d7971602d8c0ea52543d16153019", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "022ac86a714224c40cf3397c53f447584a53394baa275d223b1fd592e7915ef4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ef8bb30c3ba3181f1dd1f04aeeee5e33f74b824a4c6b989f405239d32ddd260c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c294b58891788dbf6c0c86f37655f994db9c3b68107d0f58608319e32caf2f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b9cc2da5a7787af049681079fe691c3f5058175410fa5505ad72d964795f2ff4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7c883e39afe1aedbd3ace93599bd89d7f5a46d7b3b7f5f14dea4d6bcfe5357fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e1b7ef697c1c44248d907146ab13017cdf29c0901e35e4c903f4610f25e119b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "826a9df488618b1fa954403977562d6dbfb9ac7a5559ced4cb9e3c9757d9e7d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e509cfdcf1082e68f0062dfc8fc36ac38801beefb8c1422d1b1b10bfb8eba352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f0af98bdb57a4648d7582482414a7bfe657138808dd1effcd34c1da20cd3eb8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1405cb40ba500384fd48f99ef6305e15a262bc3125b6148d69707a221e1efcf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "578c055f278bc48deabbc732861bdb1124c693d9c7bf52728941a6f4778f8131", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b131c6be4e0c3d9a98719e93e234970d969a3c945a31530e9aa7448b260721bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "16040a54e19433d13f3202ad8e838658b859537e70ab68de3c1c1d3827912386", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "79f135ff79cc8533622bfa28cabf4dc6405554cc4973a385ac5f4f20e40b5817", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5da0932e0ce9c60bd2cc0a7984af5aead1ea3bddf1925a2f8ebe948d47f34cf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e79d7372a3b8d6f0b069f06ecc4d2fb06826ec8c507875377ffc08f125dc379", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b317d4dbf807b24eb53b8a541052cfd5584d8be0ff4e696c615de97c39a74df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1593a8cb4c087c1b71ee0378e199a740a6bff8472a8de3986ec86ac7b50c5010", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6876736463bb68a9af33c16d98737af367a1d7525817416b8fde5f698f2af506", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8f0752f148d1be78bfe55196cf7dec3f1af836d06742e3a791f2a2521dad96de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e32ee57afc342ad5f97fc0ca48e9fbc635fe014ef9e1105653f17a91e0693d50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cbbd90982e5ad2802aaebd162b21e3693106a34dba135f2b07f07ef3a1317f60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99ed3392ccbb501b96afeed1988fd1398e461a2f9438698e14ba088a0995fc5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "493e8961a320cd3cac54dae5dc97b31c33007d6a8832c8c02af2564b1f1e3bb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0c3b607bb09775a29a08e77168e9f9b0cc054fa56923857a70844a5de4ab0209", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48db71f3078c6bf63aed7aa9b3e7bcf47ee1c11004d3f053abc64daef685f219", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cb46e1514e6ef9ed0ff05bda8da197179bec28ae15d182e9a4882b3460e18edf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "777587f4ebe40ec0ea95274c08894dbeb482162609899e319e7b7b3c7dfd3663", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4ca90cda35f336e06b8d697a11e0853bb2980f691946bff955e79885e076198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1811bfbff86716223f0c6a61edbbaf686592f86d5af3130178120a6d995cfff1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d316c03fa5b7b66d0ea315bfa05bcc67e8659feaceb397b1fad07312092b44d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "e86d95e5565d8c28293d269d6f3d5533ba373df9a51c520afbe69ce1d3359894", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "9e6dd5489910721a66f228e8f76d66ca7f9fa0c3e350d0dd3449b104d87114df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "9dcd1e590cee7bb79ddc51543a2e5f71689da628853a8965b2cff4d517f75904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "0507d876a3371bfc19d8d309e64811d33e675d69a9a6fdf9ddb05c92693cf4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "23649cd5dfcf7aa4c20d8e598cb7f6c2ec033f5fb227b5ad953dfb437489d752", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "79148dc4ad1f2b60eb8ea15a405743026668adfe1340345e96120040af1ec4fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "0348152b0d879bf6f242d2609835dd8e2d4d1a9ccc135d234c94c230706c981b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "390315a3fd19953f7f47356a23b94cfc054241a6544d53c8ca79221228d110b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "690e03822000cbf4512650e519b1ddfc82e558920e78fe6e2834305ed97851e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "e5710f615fe37a6694b0cb181e3ec882723ce2183f091b554cdd676cf818bab3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "8d8d09921c79d1b1ef29ed71d7426c3dafb962899315829cfa44a680ea502db5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4d6f3d05b3d446ebfc6d577374954506ba91e320976649823a401c92cc8ad699", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "9213ebc832bbec95ade2b19942749fe2a9aa6f8a627b14cd091bce6281a138fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "75be9914fc27e60e9c8b21b52d6a450326006b3b0aba8aa5ece7fb56e1245e72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e49f3c7a9410af377776818bfa7a7cd85765c12ca5ce3a8212fb2223f58f6afc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "f41c0802d080e4465fc2fe0ccc50dab62e17a8e3404562b49c97ba5acfae0c44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e2ef3dd24d585e120d047d5ed30b1d9656e8b52e63b4f5026fcc6d0939127db0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e51c6ff959f3924006f717ec7b04cd00e3a8fc2e87b2d5f7023331eee4b7de24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "c3576af136dc072d03bdf7826fc3963bc4572b4cc3e4b17d0c71088c408371c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "861559a7821de12debdf157c2af2dd38486837faa2ab74e4c6dbc6cba4de582e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "068b40d3205e900a9175a138cab738a8ae5dfef5e7edaf8b02d0408998e482f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "bdbb1e9d9d5dfd68e2fcf2e83955cbceae9391fb342635f74e29af366166b863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "42ddaaf21ce2c41fc4617861655cbd4f8c354f9eb92d0cb12cf589c8daf8a07b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "b59d312833cd6135568d9c7c4f1d17f5e82a21f7669c832543e9af4b67218ae0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "83e6bbf60aa62e964dd6dfd198af4a20d2d18dfd047d05bbbcec944d8fcf66c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "0ccab328f6214b791a8084bef8890b65cf1aac78a663ca4be3cb0e94f4f7dd8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "a2e02376156734fb484496fbd41d1d08b83db03bbcf6360593c80a121a7628a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "ffb2762731fba84db06e6346fd52dcf2f3854271986c4bfcd352fb6ab8baba6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "0b180297e2a1bdb839f129faeebc6f62f5d3a082f3866811f9a8ba5bffaeb30e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "52b74677cbd9edd7473036e273d9a0e1128c9a3b5a96fcb34da7632da447a5b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "a9aa1c6cfbdb3bd99b7dec4b3c3a3f0dda12cc08384b3615407bafe4f5695ca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "bb796fa75142a23c053a1d036f1dc41f1ab37683ef21b39a6833ef477e87d252", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "fd633a45fe355d72190d76a4ab9115a5f393ad46225e0fe48033e46766d9bf90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "4474b2d001ca1b9320e0d900aec90aa313bbd72a34db8b9e9db98ac4428a85ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "4b5a69ea719ad4780e611745a9bb4cbf0e3ccc6d3e60c53314d63bd6931a360f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "a333172a3e77231840b0954215e1de0186498a8c17068c1faf289ef8c0df87b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "63621f6d6e592cb9d7cf3f5851d9cc37c7f5e765c9e6c252d2a10ca8f9f9f0a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "04637fe138946e2a910e13ac704a1047fcac5ae30439450bcfa7fd371e87b26f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "fb6ed6896562522919fb62596a11750da94dfd7923308404dd5f2c83818b8dd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "9006c7be99a3a653f6c4f1e76a1ce20d9b371ea44e16c5fac81994f5f8ce30e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "fca64112b053ef1f9d4a6c9244e4c06eaa8c4ccad386322cfe44e0c9603d7ca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "66b222ccb0b6a7177ba11aa05b5ea0c94f45946c4c60574164c8a384e8007c65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a70854bae5cdd5ccf6747f5670bf79128af9dd891588ab667587d142d733565a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "4b06e65c0d859566372c081593d84c2269f81c892b9d7c07bbcbb9b1bcf2c3e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "6d88226190354aececa763467eebd9cb0742acc8b16a9bf9a837a6448f6259ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "0e6049bf3ede6e1e616c71a38abb6d82422cbe86ec7e76ff8536da9cc567c5df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "8b60442adca1458391cc4b739a352a216d64d38a4051aa54e06ed2e33871f706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "12bf49f041d9a0970a5d1b5231e75669694c109e7dfe67aeffccdcd0171d1e5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "e8562d849068137cf42358dddd0cd733357e62ba2953d67bb27e4800d3cf412d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "17887ae479fff478a4e42dc450d6dea5e3b98dc5317c459c3c2bcb07367ca708", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ce5b871a22c4173453164c3afa7a543658ad660228843bf5107ee74840b1a9a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a0b819c2c3ea6793549eb5fd78edff4a4ca2b6773fa75eb54799e66b1b64d97b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "763ccede423ed7336d6023c116da68065819de1ebc7e1f64500b5d9df0cd0139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "6ffe2f15906bacc71c07987886ba9427194398c9c825f37d164a01e54b2bc70d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a1f87d13a2287f2ab536303227964df3cc2f242bf840dddd0d7dfc3a2b854e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "b8bc30b3c9c66ad2cc01667dcfd8c55eea5970d4a806591cfeb902c48b21d159", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "aa32a3b70a76148623aa4b1cd0101f1780978e882a6fa31978463f60fa48f79d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d547e8f7939bd823b7467948b126af0db53935d9e41d53da1d7e54e438457d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "741f09ab70f895a0eb6352481428e8879beab9b6a692c36338bc431a9364d0f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "894d618875cb9e335f88959a7f4e89c4d9281f8ffa0e020eae1b52c3e5276e3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "dedb490072f56b99ec4cf0364f356a5ee53dcf37cf9ce44a275fea61914badc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "546b1c423ea385a95e1204b5cf1f1918924c622cbced9c7512f93fe0be9b4c84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "8a1b4ff806f9c7b89fce2a5c439823aa5924e7e0a4efc7af24a6d2f2e3c8e37e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "2a6a8690b1e511b00bcc0289a94efbdc62ddcecd8f3f387465aad7c24c672e2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "829da44067d079f3e9f89bae52cf5123f4444fe4bb8aa4199cbe1cd0281c5809", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "283e5df0452a1e1068314a9894c61d83030fe86c9f3f39cb91b2eb8a6b8ea5ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2fd4205ade9d4263a069cf47bec8b0071273346ba3ee35827a4e69911c1c1dbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bfd0766f5b03c02a4c87f3e25af4805c19a2c8f9022e44d2bcd513c878c42eac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c789a8001ab4f64fcc3d9f3cb7f2653587d6e9056e8a422dc2b152d38949452a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ecdb93d9951278d40f120ccb91535356a04af85135abb04a523cf3e293748820", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee7839d2503661e62bd97a56c3362cc6d60d333015c4bafc4029909b5cac4985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "38eeb3b69948a4bafbc339fc83773a7342990e6bedf7a69c65a81673c7c47bbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5e8ff7b730141a757e82bc156bfa1f18ad5bdbb525af44f309fac81f891487a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2158bad4dc33df8908a5c6036a8480e7b83f08d1d1b02d8b593202faeb48e298", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "08b362b6d37fcaa02fd84470b8d3aa0ff35f8be3142d67fcc946a7f6b52055b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3591921b0bf896c5e1c3561b9f978936f81c55cdb559c9876be8d857f0782957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f5f56eb3a014e916261479dc8f26fe2e478052a3cad480cb2854a7315516518", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c84d2fb82951789d65a4ecdacf7d4c39357a42fcd5f27bedd25575abe2dbed97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "afe51bc27dd1fd82bca68175dbc09a8bdcd7a8edd6e90b56c374ee32f4ec6095", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61aba468e200853c1106f9bd93ecf1bc28673663ef78b8c4cb6b0454e6ab49c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "554fa4e94b549ea7d50ed2d8513e81cd0cda6794f3fabaf5968d6f847fe460b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "be451da981414bd3432bca920363fc3444b26fe5b915cc870a5b266fcde90d2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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, "34c0a89eb740baba2cca8a2fb8252a8f61705f95a7d7089fe747b7ce6b3320dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5506e087aa1d9b9666b7b8b629467d50f9e048903b1f9f0b69449d34f42d9452", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "74eeb446d7840eff274abd5f377c8de3deece2dacae0258a67c936c87d53715c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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, 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, "9fe29321bb06d6b5c1a14e168e913f6c6e76719d733d8154c00869c3d39f7437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "17e94384910887cc6a116c2c99da1f9397a1966a3cc027fee99cd84f9a3f74e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0b4aadaba8f3107208dd0c6793794408e32083c6c79d2ae4110d30e7e80803b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5eb17e6fe62661b7a6ed807e0a0082d17439e23593104ae7e4bb4165d2a22ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "61ab719d209511cb95288bbdf8d41a853cac007e9b869e0981002a1b022fce31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fd636ff35a2aac49f3872406ae59105970d57a11112e346ba859cfa03c23523a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "df20e977eed008a96d1234461df64547df5041bb65537ddd22df99e6d026d3b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "09517254383e2f9eb7f5c83be0194062adfe3b31a3ca19bf628bfcae54f09f68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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, 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, "a518e2b0fb2e355448bf3bfd1b38edb1da3bf01823850d7f87f78a1b1a6bd054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "cfe26722e56a3b00d039cdb14788d71e23089c81f314f2c72cc3c283b4897d5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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, 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, "e02c0f1e5fde26915212ba4924f32cc4f5fc4df009df239f66ff03941621675a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "f2ac7d5a80abceee6113ccb7c3d9a7fab036cab3afd999b28ce0c7602cbe09ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "d59f852134e85ebe429741d72103dfb29e025e304a807d172956a45653d03588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a2650b621c8cd1668f28ccec5ed06dd3261e38d24704a6132d47fd6e0053ec1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f4cb4de6e455c8616df714a4408b3d5b214c8f012499d4263caffcbfef264007", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "5c12eeef5a9995931b554120b5f893180d16cf1902092b22f192b4b59331ae3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "11bd74781384c7dafaaea8f1a46bff95bc139a1b32f8ddc981a39959ef56ee86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ce7ad808fbbe979511c03f5ef4d1efcb29020931511dc7949c7925fa088596f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "cfc5bd1e01bb2331318b41d60f37c39569c239fbd8bfc0eb792ea5dcba3a9814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f26291af21baee1329f6c8bbf055b8c78fb76d520cdfd8c288c3f7005b0887d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9751bce79925fad47acb4a8e0dadf354bfe7919de7a5bd4f18dabbaacdca34ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a24680685c5b22f7b8592d0e3a8daec2eee07adee8b44f28ce2d0749c43cbdf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45879d5d1fee2582f991d35118b2d891717b9bd21914e94844d2238bfbe16698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e0daa37d1251c19e937220770bd4f131192391a68e1852f2b57e99cf1af67477", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3016492f33214e56f9b9794ca50b20ff43a566a4a622af488bf09235c3e9c301", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e1cade39f45b2c363155b9c0ce97a03479192f35e185ab60bed8c1394b4ad72c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e995fbca0b58ab77cde7f6ac8532f0ea915718151f508b57f0594214faadc10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "513c1a508c1425536dc1244114844cca84e22c0738cea5f5126b27e942661bfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2a7e3286a7a022cf49ef50e5ee3ee3cc439474aa3e4a195a973a13355112d28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "48e72c81e6a943aba35865c27ac0bb72d6c184d613bf05ff095c3124d18ed773", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d29bb845542c15a77af3b395d4bf004267a7bdad63fc584143d9a03552899a6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7ec055361bfd06647e4a50d4f8f1fa46751b4f20415b48c1a0805a26a039303", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bd61a4f2e8adea45837f4ed8a87177d71757f1330c8c5f0cb92eb7a41ff979bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5578a1b152170c3e746c73e2ea8e87353a159f3f615524b607ca8140ff469248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d02a275ddaa01c5fded8395c052871368b581318b34c6a101037a6a8d2db785a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "663dd975e354bf6b81e4b2fade59b84f76d2fbad53138d8532ab151d3b91231e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "104a4dafbee1fbe56b2d6ced1b4bc49dc7bfdb6174c9c53ca75ff7b5c780fafa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c8efa186ae2c3d2bfc46551257897d9ecb1e10f6069b353f16ea03f6aeefc51b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "e0d0b83c2b5c9ac7e952688b906af0daa22ea6c0edf7cfcd06686cf78dd8b104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "997d7142ee5d3d0700bfd7f2356fcf5720f4b56a2840272b7a11fcad1206503a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "e879b8ccb243c6dc2372aee29988d1dcbe95f03da8334dd69dcc6803cb9a71ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "674c86a07d84676ac11d2af52f8e2a6265413400c9c1ce6a3b8b93b48ef98195", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "289cec2d4f38eb3372b1c5449eb5f91d47b36f2460ebe7da309e32bd087fcd22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "502d9d3d5daff9930539bd5f151b6ff91f9696c1e447ad4f51b66a9fbac80dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f8fbae0aff6bbdc8c19a0b622038b8d31d9d7696823e42123b3f8fb9e8b3d88d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2b4b50226e264eeb114c416f8c641b53907f87a9698893d6f5c16c3f6ce9022", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "6dc08ee91c5b8e172ec3983cb8638ec9ee8b36c0e5931de57ecea6c28ec21ba5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11237db19f84b62359aa059c2c9b8889bcde279115d7b3f4c4eef61aee8fedc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99ef094f7c088f456468edf277c2030b312274e8bfb60297e5864cdec8216b2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14b19af7e6e274f2b980ca42c9bf4ee9730229ff5137c348a294d33d1ea8be39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e338c9585e674286c4c6f88319813341a6af6e48b8d881fe2b9a48d43fe59572", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "afcec28cf88934ad649e682f6bd73c6a84b90140e6dc1e5d5205c9e0b611888d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8ba566e29b86a0daee1d163455d73c3e75c95d0f020659117a108ad899a3705", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "70dbb065a92c6a83271cdcf474f939474dbbad1a8a37ea51db0d12de607d52b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "145688d8d75d54058757b7767857f6c8c07aa1b1295a1a39e961f6a9aa937960", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18d3b38c5f3177970f2cb7cf71c2b461cde477de420e55ecbe1a54d74ce7bc2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4aa251549a0594f0de577577e9963e58b1b89ce5d6e936d7c5f1d7513a7a3753", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1eb3e85e4e84245b36e15c44e456e2811e2d3a21d3a7e19e4e46dcd7ee213a03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "510b7c8d80d15a2230232d25e09d144da3bd289520932dd17257df0773ee5579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5880d5f773f3d10749c9c014ea26fba38370c7f02badb06836a9b64b2f721dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "f329aff39dfd28bd4538fefa77c1dc29869ea4bb71ab36443edcce7a5bd07a44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07531f7908ccadc86e4a5679b8b352a1f63628d88996436e0b8137c0900869f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f6a3817cbf8a250ded9e6f17c6edd6175c260b02611a2aee36e133a6777c286", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d1b54acabd8f84e93dd3fc9d593337a00922378a02a7dc16bd402f1268cc295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "736f2bca600118b8f175faf3f311152f7455465ce1839bd1271f322fe109c9c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "babc51bb5b1a5c94210fade363a6d3e0d44c07ac50ec50c9186ff5bfa46b6cee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e0b988237dd9618ef2e43b7ffe5c2a27ab8ebc25f0f3aa0a07ca329819ce56d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d25a3b8183d4c7caa005d8a2aa7dd937ba93d84d4223485bb315e3aa0b42dc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4775ee45051c28b07cf9d219b7e44623dc9f2787a448d2a8ce70a8928f0f33dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95d1c110981d902bbd5eadd61f7241fb228fe9eca2dd0cadbe957d141906075b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61d6b8f8d4658a08bdb7257864a42462a32dffc265fd5b66dc238a8ace559337", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e586b75734b0e81a79e78cd464a5744a008b0b1ac3993672d60169c6d220d509", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "398f91c4587fecda3e06593d6aee38acc14e1eefc7a9fe4da2d78159b1899842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "320049d5e3e3d5d74dfab2b0deb2486ac501a3bee8c194faec90c33004c147dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "de564b0bd8b6f3951c8e4a70ff25f7f21de2c5fb85acd2b866806de86f3503b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a3e996ee326f5f4120b88c2eaabb70a0c9cb86f682438db41c248139ecc5ce67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ea93ccebccc9f31753db3922a89f14c5d2bac90ecdc58cb19a0d542205ea904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5e44c25c4a94150db2cb2932e929f2dcbf6440365b6d19b1511ea891f57df1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f650b55c83e8ac3f710d20b92031cf2d0f68f5117514b155672bb5597e8a3c26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8b9eff2b97dac86f935cb5d3dd10cec90463b483c806981cc465372d0a828d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c6354e4a9c1304f7755f0a06c1b3ff3d05580a43eb5e868020fb24a051cfe086", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e59652f1087346e5d9a995e48603d26e28a76ba0846cecd4093beb8809caf87a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93e6fe5ba533bdcc13935c68d3fe266bacc6dd8ddbb5143ba32a67da5d8c95a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "330a63254f04ad163b53dbaba388dc84635c2c6f52d07930cfe83abf0d4389e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "59e0e0bc171a10bfaeb38ba7ecbc1391ab405675ad7ccc78bace9f8c7b3c7fe5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8d392a1118ae29393b7de62ad957964cbc687cfecad27cce0c7b2a17e915ef22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53285a3ec75325e9807c8b70eab115a0b80729600404c3d7063c5bd50c9758be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f271019a719fbcb30a7e6d49b76b11ed20f1890b5000ebcc442ea459df782827", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b167c7c6aefb155e1d7e9c93a94a15dbddc94fdd150b1659d403a82933a6f56a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e025f599fb03bfec670761facc3f8c46a1d705de5e782808cac6d53fe93bbdf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2dab11fd10afd420fe703a6a4465b6712ca64cf39d4100a06c647ec67aea5273", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40dc95e7668573a43ba047c5f217d98d70c55db4074a4c0ba78d54209ebbd3d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "183d4157b66414c7a9631cab144bd61697f0215bf159566d47c7537d365577c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "beb4a1887697e425c52043f96c22e0d3e43e777032420b0d2910c1a52bd49775", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3bc5041c2053b9c604b212cce7f6916858d1ff76485f7581ff593b08b86a36a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bf832366213b85af51344736c49e42280df22b7202b2b14d79c635094fad1480", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "002f2d89559d4dc84211d147d83db0c8321c6d516d4861d1be9a883a1ce7a185", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b320b0884ff97750d07a83267282d6067018425ecc1aafd8521a436d16ca65f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ded74a3451e9787fcee3f4db0548f25bf739bf5cf15b16802974dbd1fd3e2fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f65f7207e83533016b97d45852dfd6da73160267704ab99fa73daada186b7203", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f97d6c2f5194a5af095952d57e1e9cedc01c1b9a40241370fde4a063844153f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "71ff4db95167e56629139d0cff106adcf51a37d9baf7fbb9f64f6c6d4d9f1668", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a96c5407a89191f9dec89e673c7909cefef041dc1fe10d49f44bc74c8bd8c05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "074b7ece64116333debb50c3df3b0215e37471ea2eb83881070a716db068044b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ee929e9caddd68c36e809fd6fecd2d2c429f9b308add281b0b61d6651ccdf47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "fe21dc484f7c00ee71f75b3ec86c69eb3f061a89e30f60419f0dfadb36530d63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "695ac9db2eb33f312f367d6cbde2c331ef5bb00ae07e37f1af72dc82402486ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f60be15adc1821b672f5ad0194cefc08155472699b783705a9dff03c35cc0f5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "12d1d4654d1612866a87a511a546fdec4197d6c7a410a02a7c4c05ad6e184af6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f23f9fc0e724f9aa6b4d062e6654174e1155444a3373a58cfd79357ade762e92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f149f4fae5d67991c22ad59c063739cf9e2819c185f4005d4eee99173f21759d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49cd99808dae32f38c82fb4cc658f63443c8d2c72ad41ca6bc5d4208cd10a447", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8aefa4532155100e97847561257f601edcd27162c6b78b7e6fe7ba8b5e9c45ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bc29d8adff2da0ac20381b63c1607a4ede20b32adf33615608dc872888e73ffd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9f28b0430c7886016a1d010f57b54a8bdbd9d0817e284b34a77e575ec49cab22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e243bb4de63c1be2e9aa2b6f9389d73754bc7a354c04e73a6f6e801655acebd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "346100d2ed91c8c2d9c9776f1f092a0841102703eb987a33f6174e34bb270994", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a00dc46a4099ce2fad9ff0158974ad7502149537586aef0b73b1d6ed0a6d32d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c50482e2b4b8ac3260605dc84a944fd31fcd4280caa00e2c36fd147f822aeecc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ea0e3d8831ac4c5d5c2596303e6fa593b80617dc51874974ecf2bb1912ba13e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f044da06cff2ca29ef2c8576c238df9abe4bb7f88b540b2f6fddaff2c576ada", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d3a3c53124d0d0256dd7666534f6129c80dfcdd8151a7eee787a761f4eaf6af7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2249852dda2c60e0989afaae20071688341f1de5d4ba339c4c1f3387d0466c71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c6f776fb84a0fdf0017edf746c1db3c8c2520522c36c42f18c5e50d47debd06e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18573d72198cb3f655aa80c9cd2f5e476f288ac25374274142956f80f60ec6c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "afa0d44775a8dd71c9633aa56540908e354dbf62ef5a8ddf142cfc9aeb2cf799", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d599cb258020f15c9502aae316a7eb98cd71563789cb6dc6096a5a33e41dca6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a8b17fbf615ffb4c0a44bac18d43e9177b4d111e7849a8c1e7eacd7c2d443f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f178a46fb67341fb80afcf48a12943c8fde0e12c52f082abca769076a9c8829a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a4f86f5690927a3e4d50f3af39a19acd33814d876d59222f14531a6c171adf32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d2f7618e5ca484eccb454e9b44a81dfb0fea592161633afd7c1e593ad2767b70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca336999c62a7afd9af7e60c351fc5754671a154d13f907022a7a5921f34eda1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "af37e6c914a7200065ebf18ac86465860acbf36f938f8964d7292c8155bda5cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "369e875da43ba08b2356a187f34cfbbeb2f16e24088ffb16b96670c0028098fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "83200f3cbbae3ab0f7ab836b3e894557794334a5840298d8d6946508dd4f6aa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f826858ec6b5b919eea6731ba5890c7c55439789bf6cecf6bfbcebf0adbf3a9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0dbf26548e32703d4265c60b883817edf54a96d7eaf7f7b92bc17b2d16c8ea86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13888d46719ed007c58d2ec3db6a21598879d6a0b06010c3b745d678f35adad9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "342ae8575ca6a2657b0da6ec15009542ab7a976bf70e85d493dc02c0a7ed7f96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f26032c36b4dfc61524e89b0378784a9376f5849ad5503e99b038d1746661642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1aab49db4fcc2e50d7b066c0b296294ed5c22a8d6e84cfbc89a4923354487fff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5e3520445f63db11d4d37bf942fa136fa31c8d74809ee0a0c90ebc3115b4adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1dd7e4cc8aed734f26984259e70034758e9e4b8e1d06ddadbe1cefd8eaf5d9db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad720b92c29157c7ddaa7ac23ef6009e8cd78da80d9d3def33ff9eb2aed1f77d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b27a9c131a30d998e05b00d4f850ef90f67224877f8daaf9ee21c6cf0744d72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c425b7d546fb1e33fc9d4cea0f64cfa048d93e573c256a9b1263d09fd313c168", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "16d5e209bd3342f94c0d9b500eedb396a986f04aee7c4f3021063fe50efb6944", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0b29e0e85cc5f116e2300bf187d60a07f2dde57f2b44948b3c9482e9501c66f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b0c95cae93f9c6d37b3d3593d4f2f154790a4dba6539356db15a2a3e06ad9730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "4c9d28a0db6c73884070db145bb3f9fcd1efe8c62bfdf590ca201d8bce187b25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3236d0131c70e7c715bc2a24e9633141706795828c6eff3629986c47a098ab1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "3b9d4d056b16d61c760f0010a3d0cade84a0ac124fbcdebcf09ae0bd9def1aaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "ed4ffe15880ad275476e7a91adc9f65cca71e3090c1502e6ff53840b04bf014b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7d26e02acaefa17d3f4e6d4a7677d123b310a131407879f8e899234866df3514", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "6caee4402174685d6ec1d5d8389d7ecd520bed8eb66add82be7900c258988040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fe50ca5af60a41c0a3ec8e857e04502d4f379c7664101d70de44a800bc571d5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ab9c5b58989c4807a113e4255edf96cd9bd52820968de6dbb4bc4a99e55624ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a376d6ed937d504afdd5724638bf8aba95c2a603734b68affda98adfa0010f8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "f1c656cc5b118424e8411e70252e4f347f4273b7922d75121a94d5884c8f27d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0a4f3462e6da9643a22ec70ae2976accd5c19d423f0975dbf231fd1e28d95b9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5bf70215add5f5044e88317d0104a88ff19d4f5f378562a2df5ad4466a9f5c1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "72ee52ab327461b486f1a9c6a6cb20cf7fab41d79cec88a07dd1bebfed42c9fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e1d08d91e0c5853d9dd8730ecf3846f95d27594728297abf27cf5033189e5511", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3760d6db0c7229a7a72c4263636a58845f12e050bcee9341f9e54ca3b68cb9c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e926163883d5e431b2a40957f46bfbb3ceb8660b158761e76548e6c57804d1c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "64962b44e0e8426650a541927db397b8fe21456745531f0115bb18ac204354cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d3c737fc572eb2e67b4bc1a0167c3bfb73fc773456a872928fa90b6b784b4138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "445ffe54eda276c56df63c9040ec36168e738ddeab6c3972e3f77a17203bcd6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9902216d16a2ce89deba28a032cd92f75e4f7fb08e3c2ce1e9295f7adbfc59af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ad6b907a240edfdd3b619f3d9428d95cc610c374617683fa516124339dbf5f20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d25338f721bafdfb3ec7b9fb4a844545da8a4327471763d73d94211d35234b54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5826914be49da325240805d3ace6f318013a05eba50213406b2590ccedf8fe93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "29a34c7dd51bd35298f63839352bf1b8bd99f8fc5876ee24cf4319ba29657730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "89e89853304e46bb5b9ec2ff613e0fa266b5442326a0b9f16f1e1147355bd9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c21b84be253c434b58372be227c938618a350ba5737e84c69670215006bce34f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "152d248d07d014e8afbc4609447dbd083d2baaf71b5276cbc37038097c8dbd8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7e46d5df033dce221b80ec913926735119af6f6ea43df3a841e2208a01826202", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d0b683e3f0269e3789654a545e7ac29bd9419c3b35db6c7c69177bd6c3311e54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c1afe5e376e3d782d0c8d37404ddf72a19cf4e1e125d256993c4b90d073e109a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "226113cbd03034001662e20abaa9fdfcc36a82109f0890f08a5d760c9c6f8038", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "4f6122d9ac2d0d33f3c4e3bad2991085ae8f2a53da0c3429115137d98b503acd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "83920e9a92e47e8c7abe3aad832d4ca5ff7fec144fc6ca59762597dbe0890364", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "81c6f582a701c642d278d45254afaf0e8876837770218e9a562fc011a62d28dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e34680033566773a3abb4890de5aba7869a351522bd5841c75b6009f09654a5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c77c0b082f2549ebc9f655be29287a57f9e74ef4bf91bb6bbfb0eeeca5929f88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "422bc275b89591c0a62cff314a0a400babd0f165656efd9a113e7967b68be9f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5dc00801a12e18ed048a4127be0477299bb77c16f95a7d15784810ebdabe7d12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "fb9e8a6ceec95e10c454e907fbc3c03e157c9008e5af7f08dc7c9e541d953c60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "3fcbd669286381fbfe9eca2ffbd7587d3e6c927068e46a9f75a314a9d3b53c71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cfdd2818dc4eb6aa16d14cfccbf45666ca7b0847dbdf55ccd89e0ae50158321d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "58bed95529289e17d0886324dd9d63cd0566efab840f869989a7fbcd14bbe184", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "bfed0ad101c460c0479573bf2e934fb893c9d24fc2156b586855478bb1bc7e72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "32a4dbb9e8c30a25aa94fad35d603757ff3fe493dfa92668f319d4cea5252503", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "63566b1d11f3dfe1b49ec93e5ecc9faedabfd4d4463335112de37adeb787575d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "73abaa878421680f04b1fe11f8fd0935643c46962dbd0c47a61b7d7919ea35f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "b0897667a230c63938ee956ea3c26aaed878183ce6468c9c0ba70e7288cc9df3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "685850fe9b48ac1b733d925f456dcb09484058135a1f506c9069230cfed482c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_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, "b01863a02b1cd19b036ec387338a6c32dc2420b702c1728abb75fc2a45d810ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "b005b71354b82678b68d43b435b688010af04f8a5e91c7c49a4dcd1d51eccf94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "b0fbf86b1a800b37dfe26a4a832c730c76b66eca9e17704f9d20bc2a614d7c6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "75394570a4491e1ed038a3d5abf526273e2b3a26db6becd75310bb45aacf25c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "83c16d4fa43898579868d72aee501ce20158bccb76bfcfc4eef05b8049409c58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ef7f54ac07be31fb409c4c59ab406d4583fea208422259c34f30d7c12bd264d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ce1776b82d5acb2970d85a6c22b963ef79f03520ed6be8a0518fab6c1d5b1a60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1b30eb4bf72c4fe0bd4c1913111b14c8c7fb1bdde81ba1a6a2138a8da3bde110", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "bbeda5ab8a1c3fe543132e14ed44972f79d5030301473b0b71b6b67b73d75923", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "29239d3be98131c23173dcc90e89544eaefca138499fcbdea1a7a0ecfa5fd13d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ecc29f418f269199c1933c77350d1fb7c3b0d69c07352831ef9102f996c3c725", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "71352eda7d392ee4953bdc86eadef363d793841a9fc5a7dd44565505c7ca4b63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "20d25d630e3b14f75732226bcb0c1e42f41a2a615ed210537e053766be990a86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "3db445211fc3865863f403461a1ee88daac4a6e6d95f608ebb39072435099449", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "93fbff766912708b5380678537346b437bb27ba0f185bfff034c37b4dae5713d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9a902abd3d687f902c0553aeffb6cd79a9537dd55fc790fe3e14e38b6999dddb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9551d9c03dfc94d2d62c89997a69c91a3518660de045aeaab4e21f4b14222d73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "02d6bbc3a022a3eb01dcd5003ebde044bbe91c14c2c9f603823cf573f43232fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "682ec5a931690c8873a1fb8f7a446c1b48f2875d93de82ff83432ef3790f4f81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "02d3272f590ac30dabe8572864f2f03565882b7c6103b82e4d63a01ab9b44d5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "38031af95ddf29c81998a718d8de5a8447699d27bfeae136ccf038ed35cadc01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "5c4ab15232c1bd3a53a4a577d1dfa3a478a990fba0f42eac9cff0e2d77aed3e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "18d5989d9ab9bb4c572fffb64c8a4426fa5ba0079e1dde831ab1427b0b29a962", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0026f381cbef75da97148c834282623c6eab439a71c871476d7e0ada27d7ee6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "fdf6e3db3e34dcc54fdf1690ee77ead5d0a7b379fae4dfb499bc80d8a8131174", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "6ac1be72b0c9248e2f9b27214e8105b2b1a8c6208cfd9e5205d88709c8e6dc24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7b088ebf90a1edb56e36168495d343af77abc8f0416d7bbd8d968d91488795f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "dd5a713ed95b0fac42eaae59515f767189858fe80bd44901e4698e0750bc3208", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "2381cf000b305f21521bf1021db2f27f2d696d1c928440fd4615a549b5d394e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "3c6caa024f3dec4720de31bf139f2937f59b873228e0ed6c1e64d9dfdce5276e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e2eed15f68904b4c687acc138603b89da257dc4f4a0de2ddc43b439ee5d7f247", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c9b31d862f5f9926c824172498efa6a5460a0707e35723f4fe05fb50aeea886f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4566b9af1a4f263bc11ba68459ec73541c9208322af24b105155939ab46b14e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "b76d296d9a459ef9fc53cd25a1a98a28bc8df0de70a69f7bdb5a64b364d47100", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0a8483bef1f97915234c7ace7045dce7cbdbce93dad730f537ea860eeb00c900", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "09156fd0cc3009e516cba6fb1ce4ae324f8554ae60384032febb335ca423847d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "3ed7c03d73511054af9a1b2dbf3aeeee31178105d2fef73c7faaa73bb89561b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c32d6474d26f92c2c6e0716184eeeff88f5e92bf8d797a80220e40ff7595789c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "653969f7f04804e845489742f40877f255f05c31c51cabd7c9014323eb6cd01d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "b9b53b97f0cd0f069e430b0c95001c533a72a178d4436895ca576c83fc627088", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "e60bce67e05c3cdac195281260d85cc755b32cdd1a3f78892b70b80ac50bf909", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "4c161d5b3ff569623a15cd2f8c686df39ff492a115febf5113f813c59a207804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "c78c2bc3ca5462371f736146a7e5adcfae93a64f5bc840e1986792db9360ffa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "39ccc7d0a4aa41fe0c6ccd650ea00a0480e0892607a472f7ecc21c7cccbe0929", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3282bdf8d5281eb2516086cd3ae3bf91c557f0db6c62e3ffd944005a11c379d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "55ce8fdb039190e24a909424e5cc984cc42c3dbfd39e7c5e4859a6c46717b63e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0ac198b0c039f66040195d977572424cbd11428b8608138d88373a22d63994ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "5c4f8a253e161075926edf994d6d7ac810cb18fcf0c36140fd3d5524cb357c7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a1e1ae2236da211a04b261cf61e535a4e8f20170c0c554cf62a84fc8d4d84bc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "323d15cbd19c21d351a0e3f19595903a9196761fa6bd86aaf782d6e0e2560f7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6b8ba2896db1b9be6af09f2f24412ec1b6a8dd4236ce763bca35755520487b7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fceb272abc4758e0c704937abdb1150099f3ee25fdf84c832b3ada00071d04d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e255491a36a6800e1052f951ef85984b66a1f5807cd6bb9a7034a4ffafb113b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "17d93c3ab2b15e3f6482957dd73ec9a1d89eb1f73fb4a2230a4c1f3946c4e2be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e23be20782186b7b762df0dbda49f352671fdeef29eae3dba2f2f6a73a5f10c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ec5c1bca4cf8ba5ea47d157c0ef0c7f3b14ad246c5e16130d1a81d2a8b51a97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5368c125744c368adb2928b2a1cd37592d49e578e2778048b4a7c67aebf32c36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b1249437f711020ce80f4dd1baf82894c270ce133be0b06c4d9aa373d372625b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ffbe93189a71cc5c918faff29f6a175688a927e90ca0da4554d375ef24727ecb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b71d1bd5a45dd94b38b9d7f4f9b3e7cf3d37b081b4d2365585b4bfebe317d81e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f52dd6c571b0c1d7dfe6ddce59f59ae9e581b4c6188b5e0bd6363897617db2e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9e1a3b515ba94a2278307456522ef04a8ba3af1a9218c9fef6cf7a9e88f4e1af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0050a55329d21de06c3d4489d8ceb3b0154abb45476c080feffdc801203d66ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9fae784d19136feca137446a04abd630f70ba79991c2ee04396b8a255ae150a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ebf1cf68e87371d06454fd565a1746704d1fff85afe1ccf581801a1fc7b0da8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45201ee5d1d91b8f3d52dcb3ee170fb056aaf63321b9de8b0901fe7990ed2c82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40a674f2fa62a51e826c4840cc5d70ab62b9bd3dcf63505f19f5a0022fa6b42f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "37a3dda5eb0b27822d21053fa486a40fd046dd434848d0f697d728b63b443b47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a5915322e07848e4485715c9a0cd3fe327bcefb8d0e35c0da311da6bb073bb07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "52d51f79d9bb1ac02e9393ca529899f5188338b788feca772510d3f8db2b82b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a624c91d6a1c0499d37749e210f8c127ba69b8192fa5e5fba3a7b39051c8d0eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "efeba32aad6e8164fc3cd81c258e5a9d740869029437c64a928b7774609f0935", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3cf4301d4d8282e14fd19b2745049fe544508318c944a7b4ac19fd43a5a8d97a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8684d35d1702052076ddae5327c15c19c455cb88fd71ebd817dbe98eaba81eef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fd93d935ddd88f6f2b944d4ce3fa1f2a6a28650b6dc556d244df3179be0a48a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cc91d91075e70d34c58288f13211286e0234da5976198a4bb18989943cca4344", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5433ecf428a625074602294137f364a218167ae5b7fbabcd2d3c5e158a7f711d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "28f813b5412e00672b7240ae4c332de0d22be677dcfb7a822514697f499f0634", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "03172f1f3d712150e6922f57e770e1eb334493492fe0eeedff3ad5ed194609f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c6f9504d96022a85f8e0efb57db8a3c7688f7ab5e9cbfa25fd7af1b99e37bbf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45943bada0206355cca68605b85a815d0ca9f2696281ef14d2fe71fd49eb5a36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07f89e00694ebb483880d0bcec01562a2bda58259da0b9df0fe85cb610aad4ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ce16f468d194e838309087e9c3569e6f0ffb6a230024450b21a04e9b59f5e7fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "78f4a600a58812cbadac947b8b68d50ba13b5e2660269c46849cb8d8401ea553", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "37ac99021126192e9d19fda5980ffea5117dc374e562a54e6c338b6ca664f41d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a37d8b79dec583a191d46bc813ce66333acbbf441dfc70f49aae22bccc6b9d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20e61957ab3db9cc3c7bd4a5782a1ac8e7208f6eb7ac68dd3ae1b9fd3ef61c37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1dfea1d8c97ff384ebd047a7380a37f9f425fdc31ff7fa689de52e635f558e31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64d50c507ff1d5a4678af2b30597c58785a8d2e90fc5c218b76a1e6227dde62e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "269c8546053ed1e01c7a1a8497adde49aac030fb15dc0284d56cbf683ac94e34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "5359ca95ff7c91ad9608540630db6ec9bf11d423777ef302f554000d9a793849", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d1a60c10b6068c806ea81a5509b358541f114d47ca59490f8254363547638a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f4bdf880b519ed75954700f774868c87a8765bd18e48d4615cc4f5d661ddbe5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "df986f40f1d5c29fd591f31033cf37b1def159d8c07a6dc27f36ad74edd03c39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d99127818cac90f1fd583e07f74b2038eff8ccf0ea46a71144fabfe34f5b4b5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "c82de5aa82c10ca339a27b6466910e3f62f435d4ff1ddeb073106508d36b5446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "094315d38c3de314d5d90be292c613ec2fcc70bcc7b02d05eb787712c12e9726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a425d7b14ee2af6fd8ee252b432f1a97eb88932155317197675606667cfee265", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b73f42ec3ac9f2f5b7dc861da1da006dc53c1ede228e7f17c5b14077d99f1f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "124a2c8de56a6ae8cf513aa9da2c22ecfd9e0c01a379b5b127a04e709651178e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "02ed8d8482866544b2c7711c02d3cb55b0601499435085e8f4b457c9f03f7881", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "002f50e72cd0b5e823eff65c793d90a6ffbfb1d3b0285c58a2fe065e61bf7833", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c3e533cfa46e96ee73fa82b5b5edc03dab9f928d27e35a045002257769deae2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d70dde89317a8b0bf30520841db3babb20ea139dc9236c897759f6360ae6cd2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2011b3f575e1a28b9d34f34de48e45482c8ef8784a7c0710dbff952a34270eed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39a350ad4f188f2c06cf6d039083bfb122e3c93738fc37221b5d12f2ce82eb02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64319a851e6b4b49ee9fdeda8d2628f4afa39320f9598a9b52d219e563f0795d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4f8c68473d5ee35c6ef84557b1b7a69164b6a8fa8dc36d4399c63539488331ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2981d584215e6a60f307c37ca78ac6f224718b895ce304025ef607665af6bf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fc124e3b4dc2bd810dcf904fa5b92428e00e3793698cf3032449d4759e8372a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ab04f38d0aa7a48ccaadb8aea6f8a7cdc432309b512be3c8f7e33046e1366eca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43c0547921d7254f165f7eca4adcfbf81d83e1eab76f94a7d589513f26aafc8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cd370234776eccc126302fba6c84f214c2cd48508007ea1532aa6f9312097b01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f2665afbdbe941f01d929802e8ca49c3716ef11a4d1f9dc3be6714f16818b64a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "28b61c2452dcd3aed73f2fae153722a517bf8473c6f48693ac58544fa91b3f68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a3aaeadc8d90578408d0b2ee2c249f1044aa9d0c5ad56f7818b06f1aad553a10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "284952f2c63911af0e4f3960ad0bcd4a44b19b654cd4edeb2a92385790dc0119", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3970c39255d602bddb4a8545ad5dbaffa13644a0097e0dbc6cf31d582f8d42fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55987ac5b006b99d221dd5fb2a8cc70efaf512eec7441ede2fa0ac2685949de2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27d28ed2769ed4ae58de94b3551c6b8a3840be114d64ffb12fa56721c3c00d55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11885be12b1c4b68ddb9ea2aee5ac6132bec7015d31d21ee8c3131179ac7bab1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "96b886ad0a337ddbc68827a55173845864b96994fd34a37341c41fa6034fac81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2b18d4f3e337ca5fd2aca0ee0162174292f5fabce0b0de515af10dca37b76411", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b8b9726e7f156aa8957fdf699acb180633825508bc4d3e034216fd811ea0b86b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c9ba196f7b9e319edea8ac734e9373b9aebfc2fedad4414da55d38af7f6f65d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2f2da4279041aec8e7f302f3fb5f14140b26f596c93498b2efedb2604a90ae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "22b50aa693a029cf154b8b2fbfa33f02e82b1637f8b7b7f2f390281dc0eb1b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f05a04f964654b223c9619b7b021070bcbe634f79a60eb7bdc18fa0278e94c9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "22790b4289461632d0394e4e98a137de9fd19fb23d441053ab5c0ee7890a6dff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d0fab10da2ea9672e338975c59ce8f8df58b988bfa23e6cedcaf37969e15277", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dc030841d7c5ea072f2ff207a0cf04d9fa71f208bf0d49e046c0980ff55987d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49414c2d09c151d3eb41ccece384ee50f35e9432cce54ac8b829c1d7af164cc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ccd9055342ebca8643dc87ce16efaadfd2f8f46ebf059f94a4580f7e0dc19364", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "73a0f7a8a40cf5f9b9cc85bdb652ea9e070596d58b8ecfc72628e44aa2795004", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9e0a702367c8d0714c93905b5056ba53b4b279c5a131f47dbc6e217c61e0c78a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e650967190178b13acfe60c82b5ce8af29c2063788c4d14ead1d5f15b7666282", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_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, "2c708a1504ecbe52e61e7477e9b1b047d84cb26ea06c6d20394f2bf7f1084b9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8fae4f607c7287f93051407629de52b9f3367fd7ff2203276e80be03a2221e38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_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, "c619b8662cbd3d02796eb6c48ce0ff88f8b17d3ee1da749f8e5cfe966d7f7e11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "554e36b9e41617492177a2d169589cdaed374a32e47bf7938432e1708c3bdb6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3a596bd79dc95053693426d7616cdbd2a1e7959f608e3349c2d22e69b224bd8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "5a79a873809786f9c9110c570292f138d6fcc7d94872f647a26951ab68739bb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "2b40d4e772cfd059f19636f5e46404f196bf42b61df7dc6b9f60d4e73ce1530a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "b9ba753dacbe3c5a4b86dd6810f586715d8e41c2494091e7804a4d7d46be3e85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "557a32e9863be83ea568575e3b40a3bf14e5866fd9a302ac4afbec2a32a0afac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5bffdf72aad8e7f3187be3b1da32d494e78b7b5b1cd8abb30f187fc01731165f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "8dd2524f8cdad98bddf7e836976ffdd2183ad1a467ba789a7a16abe3aa661166", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "6c9a8aa1c454a684689aa540fef2209dce69afa7743c1e8063aa125f8be8b5b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5dae6ce855e6d1918856905c1dcdee09ac908722f18ac4f4a61ccc66fddacd33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "be19bffbad02f07a4a3b62f39125dd13f3bc6bc56402be8d1e5667427c665d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "b244df25ca907cfdc6edcae30c434906950535669494d0283eb37b851b39bd9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "24deb532f346f2d58568e1829e002ceda49abdbf8abb9b44b36218ca63182a6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "78dc96741f527de7f5ff89ba736ef980dd3c2462b61ce75277978907cf298805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fd2ddc41e9799ace70fbad81380490e3461410e7bb4e90d062a1303eb362b9c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7322ed877861424a49bad21349b4dc8efcea53ce055b9c4b5d169ca5701f9b30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "6e20fdb0103a9abcd99e66a13cd50acb9cb3867198c9fcc34ce513cd90bad500", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "2d6157301327c580d12cc4a1089d7b01d873926d62034e5d70af1a47b6caa87e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "30362cc21fb513c8b9d8d5ffd91ab5f8224cb8fbddbe6d199df2ec441f996bdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0f127521e8fd52791bac8e9652c4d751073338b27e9e58a4a7420bca9c86e27e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9ad6f467fe736ee36962d9dbe0b4e0cc0316ea6a77426f14dfb4a60eecfc356f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "34c3b1a2321a052c4caa46fa6959ac9853a04eb199460f5d781b129222816776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "f69150c687b514c3cb8123e6d38caa5ceb678aa0ff5ac9acc5e6d251a9908de8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c59a1c6b5b0be209835f1a3873ab570aaf61265a69a2059900ccfa0039983328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "486c78de47e59c775adace1eef5c3444060b5ddbe4b860c0a559e0b07ae6beb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9e29ef9c1dc33fb80b177695bd0e12dcb6958970c8630c5f72329e0556b4f9dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "91aa218bef93e84db5de47ee372036f1fcf9cba0c0ac05c789e32e64d9209fa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0e6bb6fc9ee9653f3811499c0dd289714b988a309f5514246140ee79a08fe0e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0f10e5f44346e5ea11703f453ca40d28dfe9545bc113742178981608f7040c94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "246f68d4d16c7c82e0fb7b09ce76c93787f1fd3e2fe5180de92a961145aee58e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7bf1afe82c592a70637108b7feab04533dd04acc6cbee8f9980f244aef32bc83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "6d596f458bfca9a3fc0954352172ba93f0bcfbf9b7433cc96a316be4e3f0049b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "aead9814578118f9591565d15d028c5041ab583d448700e28e108522ca0daf32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "045674e33a1cf6244947accfb6841d079f5b7f65338e239f21babc91d84214a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ec58c7de58bb79fd0544a3d521498bf227dd7eb361d924933a78a8d022ef5c83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "d8927655ee55a32e6eb8551de43da7b18c0c6c3531d42685a2fee791ffea71fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_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, "ae4498e980fba6259f78f5e6db50f8fb229f88f9f481eee83bab036270a2dd2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "c650f441ea69114590266fa09465cd505cbb38af8c4b87f568c3d03891b66f52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "8edc01f28f83333bb0e5b8088128f240418aa6032ea86fe369f13559269a23e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "9876fdb522606c1776ed0f8c8a1f1029dacb2a90d07ce2d2f4e3848675493c52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "0796a6783564807a99ae0037c215c2c5682d1a3b7a589d1ff95cb1490f082c2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "85e5c377c2aafd3955028d691b7121a99761d5c0f03981057ee56b2e0ab5ed67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b15424de4468e6bc40ff716274632d8726bffcfe085a263bf1bdf4294bf61d9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "7ad1538269a0613b06304d236dc93cdc52290ceb1fd28f7594f3192d4f19906d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "03bf9c40925b9e8bca0e50f102e86976828be37a5cb4a792d19a13f9ab08a15c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "8d02540dc77abbe33495f7e96826de21b81dab3e0b56620a01d4210eceb48af3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "9049746afa2a676b9b351f91a4d7a58aafa77fdea51f9eaf191ee7628b668d6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a0b425d1a1b9dfac715635523f2198aca98f219592e1ca14556556ba6ce12477", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ffea7f12b36a0c3c674426573dd679aec3699a3dff759f74927d196c416d14ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "50f0cf019659fb8957662875c8e7a626d85470afd6ad8bc0075edc6492a1e993", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "0fd422cb43d270e3882831a368d003d6b0b82ef5b73f4791ec5f22a6edbac4c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8b6b20829611ee5f85075d827cceb7426492eae3fc5e0d51fa9b7fe0c91ebe09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0bb443add6a78018f38f39c8adff931fd1a53abb4828c8d8af0f3376cafe327a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c32af42ef0d7b1c89efe8856248b44bce4da9cac332a974edd3de6579d07392b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "653e47446eb0be47d55c559ac6d7613bb328283fa2151a02e36b16ae5b3b1dd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "65fd4fd6fff5ac890791a07564256916d45258333e718df01e1bde2b05a43ca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "47d903fe0dc5139140c4103f43744e05e251dfb52a4407acdfc9489ba69a5c20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "661dc552713457ad60e9c84b4dab96bc42069606df13cb766a035cb90df17579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "eeda753cfa27b50da3c9062a14063655c092b38a0118d455f1efbbc431112a27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7f85527e99f932e8b72cfb55e322510eb6c869481027c5c00d787d9e6d027664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1bc9f149a6875cbb45d7af910d2894733dfb6ca1b44884e145e057b1cf517490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "cf7f59f11f23c745565897392a1e1c6a2d11f0c18d93c3a57b05c046776d0374", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e41eb687c7c45f8b19f0022303fe2b6f943e4865b19a120a5ba8cb83dc830b04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c1a223666c85f347509b764cd3ec28cc77db43776c0d77d28a2da53a4f81aa4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4fdd18b0d223404c18b34a09f025a6c4ff3a041120ba03f916c7fd7b6e017726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "afe219488912fd0d76d13cac694164843b5cc392a3882bb73785f3dab1f34d79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "38457750f898df9badd3cddf49feeabd26597ba02ea7698886fd6aea019888f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9c469e04918f71e441c8c0e1f80c4dc44cf138da2c6a8791d0e37dc70104bae6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "5d04b1158da2ad34acce11fd9ca49a78bdb9fc4c411120c6ca144774d003d85b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f03f66adc6d9dce7cb3cc28c7ebe6415e9924e59314ef27b839ef59f4cea23db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1983777f44939b8261d1c5eb8452adfa2129360a7600d9c75638c63a9a301246", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "825e6e1c0708441bd1c2d683871a5244e8a88d1906f63891d3234405fba97380", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e5c97e6e5a2bde135ffd758d6c40114272f5a33034444275924da3f573064820", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "31cbc6da461e591b307f5e2c764b5a24e954a1d8797eaabe75f20faedcbba0ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "6b7e556f7cf3308c5d3c8c3ae3186354c1f8a3cb984a0015447c039807e376b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "387370c9255ff54ae26649ced81237931af75fc07f16c26896666a3628b6ba48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "6b18c3c524673cfc15a20c94fe047dfae198d2d5a6d02df0db54a56050142493", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "60b515bf3fc3d377847f587de446235d69d675d9e2ae472edfcf6b8c74ea4acd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1012e7cc482781f44400dbf44708fadbcd5b30894eb2496d89e6d89c7d6deafe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "619ac45eddcb27633f71e6316fef88a78d343baa9dae2ee54662df424073b5d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3182a072758ff11551e46aa07edc83bcc1380919136769f8865a6ee3bf16f635", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "d45dd3e3e9fc7c72757097352b513eeaf71a294a29aa7f9c9154160214ef1850", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a32f4296a747a66daa3b2bf243d7e19968b9f77e8f768688f8dee84b866ff5e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "9d9f94e989fd86bdc60bd493ba31154219354d502366d9f3ae8913c0cdf5939f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "43d255ef1a882b9d912aa34dd33e409fa5c7e0b2bdfb6efd0b475fe2c24d6fc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "dafe58dc7a3b1ad777f8334e2f623a736dbf404444d7bd77475c9b3f09bb63ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "7cfe92367b86d19f2c282cab7daffe00893d5a9fc5037740188a2b75e6551e17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "c9ebff01ae6e82fe2e808ef855f05b7bc8424e89f22a7091eb48671219706099", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "730c478b20ae5cd87a4fb894bbd2cf75010ad9427f5a1732ee26084fd87de9e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "62054b0e3f1523caf781ba4eb3586f93d373b2c27de5e3689312092bfc6a8177", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "521a225838c272e940e21d145c99cd4c945ccff233334c3ba8244912d130e501", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5764aab9e159df613ebd539e5f9001c61720c01f2ef315bc8af55a6a3e3860a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f63d1cbfdae5c0f338b9624682ea5167bd22bce178c2c98aa60746a6d35a9694", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b49cf4e7f1e10b857a56c59afb5821999cd3ae6771eda36410c9e6111ac6b23a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "42c6d434b569f54e1bbdeba0c0c58ed6c5b711a00e4c42a9b4124743192c59de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94fe6856a94aa23cbc0b95082515a59e42626a82fc99580bef8b83955e351b08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "95a6f26c80014d11f59ac58d389c96712020698bd6564ee3d069b41bb3e75891", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0b0ad8a509b23c7647ffa310623b091fd18f02a0ea74880fcd08a9266aba5218", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "58c4ea6b3221856c033a0d21632e0cfd3cc99d04f8f149780397860edc3a59d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3702b8e2d98ccc49cde799389cba05eeac0b72df9d57e732a15f96aa22e4dd16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "02722381c8cee91c7936cc9443de341a744d166920f33adfcf4126963eb091eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2128d71df085a19b7e02a7096e2990a403f023a360d41df9ceafdbbd8c65768d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca59c211c153ed27bb9b7564ef9c9e5fa01f97e0593f202dcd0f1c6fcbb601f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45d0315e51dba604dfc9c0d2aca15750c8d5f07e2ab4128026dd88bb0f96a824", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bad4ee976fa3c47f6276dd953a9334346c8880f1f04c859497281167322f50af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9c660abb3cb41fd26f721ad58fe3531936b72b07b5163e51a14730e70543d82a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2b1c7fe0020babe2d57c3f89b218d72049821340ca889be985f8a63a24dca7c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21df22e28712dbbbae201c7e5aa8a1a374542218fdee0ad3cb1bb6dd6c692773", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9dcdba137546614199c4b0e81aaa598801a3ca20c215c8d3ce9500b028a1c3f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7e8251ff90560eb8491918c2d20ef5c8d5aadf53d026c59cc4b527d26371f7d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da053faf5657065040b78bb0a814ae3a2f65e2c7ef8c6752b84af821d8008882", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39923f9bae3828422d2735148dab0356abc02d2b83ceda023f42ac1e3b494664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "33ddeb2f8a4750cafcb5a00662f24642041c6729805e3213376e465707e2746d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aff95516f1483728c73f2409aa9769b76102676c656a9f9fa577fe54fe5a6050", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e7086c17538d5eef220f2c236eac16a227b922a7f210e5c76bfe79f180fec42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18d73bb0b01468c48393a5a65bd67eb9be9bb3d39af7bdf6e2f14c7dd4c71569", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2bcec8b3f766e9ed4f37fc8b007049ddec67d9acf9cc053e39c0a32092d1f2d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a7dbff932afacb18e3b3e07a4f0285c3d7b415ddaf6450b83227d0de05a5da5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "259c9ced3b64bf93edad1210956b44faadfba9255a8354de90d3f08b301edef0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8cdc201997156b421855c027069bea9bc0a83c95e39633123909edb16d1b1bd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fab97283524865a1e58f6bfe12e5fc07d3041ac3e99600e1b10bb97a31a1f663", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec741f18ee0384f51ba1a6b12f2f03fccbf7e110b3740e07c3e5a36d73ae3a2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fdd6b8079f9c946769d125accc1aeeac7f85d1cc0ca4d4435056705bd7c98976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f74466c09ffdb5c0e876769302a4766bee490c4fcb2a8020e6eebc4fa55b6b92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fd70e1297bb9c298b97aa3087e19288cac57f330556db074518f612d94cc4265", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "732945b7de48edbf7f1c42388bb7894131b8d20e085eb5ee5d633a141dd16ced", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82c2f28b009afb0297cf97cf1002b5200627feb601b395e52be9c0e1cf4af917", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b33887a35558ca1015330903c13596fdf8d394ecc6636d1e0623f39d511f6139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "662fe21c3f2f4b9e5f987de883a4126352eb2eecd100010f6acbc3d4f1e35486", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d09810bd4858bf4a40f51ad07a7686cd9a4b5b97c62ce6732cf74c631aaa8a70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7abcf18adbef39b42830411645014ade78c0f99dd7625b9b08e6c5d647e3b5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bf5ec815c3c7f3b1342c7255feaf63f987b2dc3aeb245ab06890c4514a869df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e7b016e1da7697f0b819533fc3d726872f5d04ec483643f54e062d3b3d95be4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c68e96a5b43101c1639ab4f8ca6d46e97187e900ae3ce7f036b28c570aba5c69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "f5d72d2e4830bc530875aa4df4e11bafae2f50f0d79e589ac62e1c0a3aaf02a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ecc6ac49d2980862938294c92154e8c4820614bb0b01346d01dd2ca6cfbd8e61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "f06e71c5f9c99150bb6339a62fe8aa4ff623fb349a2630b65eba191407773814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8979f75059782f37dceec899c693f9bc6f6ef7b3d770fa25a4469316f9092eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5fb2c8dc1ec16742807a1c04804a936071c0b24b965ac5ffdd03ae8d25f30fef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "722606b4f165f95054e7cfacd775e5bd6668007d44baba30731b1dc77bf09fca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eb405474c25a555304cc0b2f5feb4d0aec3cd54dae35781149d168338e669c57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f32f1f9501d044daa5168293d04312e49fc8151c5b9779c19ce126f25bc8a0af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da3a7129317f7c7e0aa7486ca0ac868f587b7f5ec6ef81875f1a6ab383679afe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a5b37dd1871bb9e3412855ccdc0114f1924335db858c6572223b02efdc214d48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da75b14e229529fa0d61848ced3f63f243b7f1be2d917e0baa32f8b226a8a549", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bc482c425c38ded66add6a00c345096ac24418eac9bf5643e4a7a5425f5bcaa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dd2db8621f27615934106b9a31569753cd25309113064eb6cf5135408aa44cdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5620b3a2af7259269929b1b117b73e421f019a9a73056871bc61e233d6f01260", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3292f5862a5822e6c0eda067a52666a2d2a8b2015766a7f09055fc1dc4bfbec8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c7996d1b19f15ce9f078695a010425631a2b7e36821bcc96ef6893326d4911a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "77536c7bce18442da076824f9a387b2d89f3e3b8a36b8c337008f6665d953221", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "31eaea18925daaeaed92c4073ce811f9acb1181715bd972d2f16e5373eab534b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5343f96d8f3826be800af5590b96534963d9d7b156fd1107e269c6d53c858cdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f07fa99b9395f5a59c06c0790dc89b9773ef1a70065422b834a8f77f2e80aad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e4af4d891ae26115e6b6ffb4ddc24a88c0589232295c7f99ea5ebdd734a4556", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "336159e70a867decc01cf17d213dbdd3ab70ede214dd05b995961161ba69e7de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bd132d2612bc8145de96508e73925836574fe129dc514556c3ff9e298ad223f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c7a3a0f1fdeb1457f1e14d762be5d656d6a0971d5e7e8da7629f2ed0bee0191", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "beec5750b8a8e2db5e689f8bf1b612213fef8d848f63d164dca3dfcd124f9782", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "354ea36ed20c383589e27739f9793a2926597ac3a570d7503fa64f75d6842625", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7bdd9906ca23e8a89fc645f8fa5ee1e8c1ba2c1540c028c2e35ba4626dd2fae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "533f371ecabe533698b275a349b031be6451e75ba0130ff5d174866013b23dbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "22dd28166fdebfe73dcfa0f98163994da078732f9cb118017f2f359e039e664e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1907254e239cc2948af64eabe539ce40ea8f2a17cacd63b44ffda4e16390375e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "38710dae42b61a8e9d7518c38f8ee4890d317c7677a7f299297154e29db8c00b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "113790991a62042c2caa87f189e01dc8e1c2ad8d0abfdee435a0ccc65e5bbbab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ec8323386add255997f5283fc68736526204774981f9510014a994e54d021c39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "74be01dbddcddddf225282a265030c28cde65d3028453dfed878a1aa34416b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0647474c78fc0c01c7a614430ebf922631fde78349296078181224961d57bd6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b61bd5d7482617e001f49ea678434e563dca4d8b0d46b3dc6408cbc314468728", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0bdc0f47adb8c5a0153f6abada61e3ff0bf63410027559a013b9133dc215a06d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1260d0123551938229a09a523a9f116890400e7a155a2b3cfdb25cf5ee876d63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca80a7963ae2c61c8794958b28a9c0c5144981b22ff16facf34e75fd9c2ffbfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b99b94b6babae48ce1ce46893e6d7311cd35c27b6617fecc6c0154cd15ec56ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7163b3d3bde26fb808a976b3ba9b6c0e0402dccb8d440c914639cdb6fb068771", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_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, "d7e5a1e11361a4b80d03ba6e6b7bacd00f844e57f72882796d6007381603f70c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "256ab69409dbf17b7efb304c6dad2654213784d8fdd1601b26fba44a91e00bcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "aaf335b41ff17160ee60a8e818a9a8461f5d4178ad483aeebf4c5a75e7a794a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "dfdf303fd55416b0e65f401d9890d3eec3a459e8a766ebe2f4d56938077fff23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "840c6423c3df841dff33856d4a8bbd8d527b406a8d5041e387d6afb2ea02553e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "59679fd75bc2c0cbc09d82bf53b2badb4915807001e18faaab754667779aff18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "f609991e27cadbd1962e6b692aafcd67068ef4fa082183081f2f35c85f951973", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "af6f64efc21829db93b5b423d21cbfcb53f094e86acb2e853dfc6af118d24c20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "143811786caf4de2a1db0bde0fe118e380f7596cbc7a2654aaba81ce9d6b6b8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "21cdba810c7d76c6299b4e482d7d7e9279d6d93ccd02b75478a106cfd06a953e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "24828bfd6ce563e5bdc0f4f0be3c7ee4b49f623820edcef9557483686bdbd05e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0c305bc8a7149ea386372ceaa24196109c1a5107c5cebf684d11a92535c6ddab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "85244c8b7943308c1958d0284c70b060a06dcb8304058c52e4b0e493ae466db0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "62c3839b060661d99f421cadf6bca89b4f63bfc227962c07b6a2f17c98539d26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fe28eba2e30e83aea9a668ffafb121fa5aad9a060b953726b3471b529a208cd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3be5f9918e830bba2bb355c358cd75617ef467614cf935894898fa4b5c9e8832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "4b94f809205a37855cef03df43290930c3f93e5216e2505e60aae48fbb4e2d12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "5cb59daf455aacd50520c3b23f8095c47913efb5cade5a9e854a147e68a9eab2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d13436dcc5dab4ee4c039490d5ec0acb2b354232b6ac0a79fb7d640f3bce997c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "2a2ced5fb8f068c3af2772dabb2f4f4ff3c72d45ab5c0b65ca3838088c34ce68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "4c7c22e5c8767ff6a045b45469ba8b90010c7078f8ea596e37d7f420f77977c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d3e4996259e2d72cb9fc1f70883fa612c8a7df62183db01f05828a2066f39fa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e035ff9610563335cf3ad670588d6ee8c0e66d5543ba5a4742dc413236fc0168", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9a462cfe6c09efd01804fe9f3824c5cd57458e9d51fb557c158c2e2bcd85678c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9be656697681137b154ccde8751ff73e314d5b95768ba3132a7c4748e9559ff7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c5b6e030084cf67b88635eef1bd6aec86f4439ae50caed1229a9149fcdb39f2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ebcf7a99ae017c596504982731bb61b0ffb4574c3666a300058a54f8adc475db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9d504b8b9520c72b74a286fc02f559e5263dcaea1f5509ae715d79d50345aac5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a4a9bb1dd40e40b54fb9f9c65d3bac3e64f4bcfbe2a045e6034edadf8dc411f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "21131a21561aee726e378cfefa06ea36c88f5431d92093b2593b3d55e2dbb771", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "024346c360282bcf98d3b40d02b7e7367878ed0bfb827ad1c58cde4459bfad49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "544bbe453126b737ea6294b8ad135eb42763aaa44300f01143f9051049093149", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "1cdc8651275145f663e86509025ce5d39481908d9b51c4cb43f28fb3c0b23160", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "3f11a8338d18f1c921e59b507a0cdfb576261d113737114e470c266a5b10f20b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "37ea3357e9f6adcf329425784af182db363bbfb5fc685d5c6a27284c78a27013", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "2518d1d4550525ad435d768cd238e2e6b777881e0bd8661939ae692266c82ef4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_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, "1215736f383c988a65604c47f4e1c0308cd2739c0619ade1be86d8ab089efb62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "1f9b5516e0d67d355920c4d54fe78f4da3bd6b88c849635032042cb03c83d484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "49cb9ddc68f4acdaf3a5b18c19bdaa7b6339e7857e3042ed9785bd4e5464ea0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "526023aaec1c2d02f7be6bbfa1cbea660efbe8b243d31f2a2c75dea29c372c4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "66c55500030d9a1fc488645a23ba3c6c46d66f6d4386f96d519a7d13c40ece6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "714e9d11ed5eb19fd351829b2ed20c2915ff3f10f6230ab2ff4c2f2f3b1830d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "6cb361224d87a648a813fdcce11dcff6fb597520011c567fa51bd317115aee5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "1e521cc0ae85e1c8b4ae4b5bcad5200f4c8efb44c1a3ac04269e7d66647abc87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "bc25fe1583d29d499c7cf905f6802eb16c327e7a8b956e5f18625af0bc43b4e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "857390516c1c885d3579e9cf948aa66446f887d45a688b0129682402f82cb64f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "5959405b10d2bcd720509c9a8d73a906699237e0be0d069f5e8c041e7c42910b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "db65439568a0147cd6afb8639302f80e152c1ba1c1caf39f4eb33b9cb1a08828", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1de19feca2d17ecd6c8b9c48e0bfb62349d4964177e7043db127d2bce5382058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7ddc6ab20fc185e49908d4f1a387a33118e57df13e5c70ac523d60c86eb13ae1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "595d2fcd4deb534bb8758936b8c2df9b2e40773189c4cbc09bc0db6549f2106a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a71c9765f6c65f5a1362fb7f7f3d35a6253b141dc2d1ed89035dac0d2e250dc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "13303ecdd2dba03cae247c19aa27a30fd20158c0b9cd6c555fc97007ced5cfdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "23c28824562d4b66f509c4b3386d35baad71941188f585ae1ad160b6d882f7f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4029d5f011c2915996eb1eefcb9240cb483353c500030bf9d66f3d7769a8d15a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "97979dd570dca2724c52ffc6c01c0d33ce6c9d9ced1b7d37c1fbab9c24bf6aec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "691eea79eae85a9224a57030b4c20db795ea8463c09062d000aecc6e002b94c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e5b37ffedf17a8f83bee4c66f718e2a898200b86d4b9ab2d29c049c34f18ece6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "fd15bbf01e027c9ce7aaf610459a249798201655bc2e3bbab02aa40826467033", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "66a1be64a8ec90766008a36d4742c9b6fe789389112f44e2f3e7a2077c611974", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "77be9a109e44d49855ac5e4df9938168a970d71a7de85b1b6d9a832198a4c65a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "42c426b120cfe0ce547bfbcb0ade5b52d87c5ad2f9c3b1d8ebdb17971ceff06b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "00d9bdb6e54a2ef5125db23a96e61e0a2dbd6d83ecb712fb531cbff0439447ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "77d7c3b4817a805c5702ad17f5d3c8803eebc806a41139dc6a49bdcce56202ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "b7c21fd6b88331c327fb9051ba43186cd09e430e02116861a1109534c6388671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "95624528840e1ae9500b6ead34b1b25bd59a1ef66c0b4fb708829bcae1f19453", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1dee107188eb5f47ad0a0d18af3149e894e971acb6b8b1f8e9b65a4aac61e04d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "19d21e51e18ee7effa206069f36454d7977e0a1bd89a69965928d41199c1b04d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0a0514456ae3123090624cccb7fa5eb27746f2fcac45f9f7aedd77c3774cdaee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0120e445bb584e24b7f019abdc4ffb6c379ad1a0265ca91e7de65fbe33e250f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "683ad6f3e0769e66a1a90eaf9a9fd9319beaec793ca81fc41d0d17eb9be63fec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "5d24ffbd706b52157df53731c7f9c964333f39b09e80547ed3847a1e078912bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e24e11e20aa0f08be6bdf85e69dbe6f5258fc1d6c9c2ec2c6426e9771984fbb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "cb0f83cfdc692ca48d23ae63cef147ef8d9ffa3bcda04dbb4db3954ec00bf490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "81c6f64d6060ab0a2eda4f5afe6a242b019f4168c19f3083224be9ea57f7890b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "bf2d6d90f692be2b8fda1c68612e7f4377e94ab645657ed5cc1d4d673672b167", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "e4a39686cfd44122d7bdcae7954d61409c2e87edc04c037763ac89e52773431f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "64aa7e1bbcc07d4ad1ff6fe9ea6137d4b21f39dac6cecd7574193c080e57732e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8aab6f9b667b39bfc7346fc848c27de6f766a85ba79b4eff520366df98106f15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "a564a0b2079cd19578fa4057c6b8b1d1b92a66c806a4f06524c69e651955f5d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "4d8918c1f8a9795e6bea148e731413074f3f5b48ce6bc323aafa5069dc1cf64e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "8ae96a12911d5afe0437850780c1271cb1da0fda574e563c1fc90f9fd5547962", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a5d237515adf38caba956b1e38dc37f20b0755438d6948c91f5c0d527ab3d445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f4c5ed3fd6d21ee4c1e2f4e3fef32a73a643aba28bdf671dec758179e343c50e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "711d9db3358da4da7d3a64be58c1351ca92feaf5225470d660f7f8adc65559c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c243696c0178f391ce761646ae234012f739135239f7c74cdb2ce2af7694df4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "6112ae6ac1a79569b98c279ca14d49e82101f3bd23340563d53a0ebd2de3694f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9150c14581b49e6e0bf35cb126c0f231c0763e27b863f5fa1378c12a8c1ac9b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2eb621b94e5fa10fe51d1a8c046bb14e1d82f42bd870a6b31422a74a795f320b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bd2f7ad1fda478ddc17ded18281fe77ccc623dee3fcc997765c9cb087bd96932", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad3b8e0fb5d5e6338130865fec4f3633a0293b46696e98f8aef4987dd61795d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f1ea163113d5ae61376c4abe8b3c2f464b943e3d06fc107bd97e63a8d17b69d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4b0951d9a658889d2a9623576ced0d88ac23916bade534619e68261792b4a4f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5de3ae6a99c0ac561d0cd27367af11e6857d05d679001cd177b49afb790894f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3f925ddb076c17e0664467147714f9fb9ed883fcb6f0d63514d516a4e4bd2f25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bbb2effb254f6f74150d3c6b3ff5b3e9792765bba5ba205e8d1aa609851e83c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14cb2d52432878841cda5cb7117fd4e7b30076f854cdb195149a3a8b8eafb6a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "97eec70f6a1538d9b80cdf42bce6474efda9761a1c2fb28cdd41d8ffc3b187f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf09b250e92a28d35dbcc445905714beee790327ca735606f652afbac6241609", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "49e5a044830c8400d1c527e38bfbc5edbc15206c8bb8cfa5e10806ab01e736dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45fb4e7fae8cc96bbc0588250c9df96ab1c2e747443ae3138626ac6c76a592a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6783dbedd1799d604bdd5571e484c4eebc6a7b3cab181f120f9c37d5f1aec24f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82b35e6d7a7e003a52cc9a25d653a0239027d4cd09e8e80b4ce37ed8f557fa4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d33bba30b3377881440d1dc12554391b454ba6829ef12b6d54dd960911a81365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7443b229675ea3c742d951ff1c5a7814b468032900658e88c121315fee31bce2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2e3c6eed2fbf96632b87eb8fdd9c08278466006c2732363f0bb109c8567ecb21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d261b1365137dc49d4956d85b35290268cadab8d8a207bac8824e75f06c265ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "64d7b3f265772db2fd795770038480776976a69a476a296768107c3f8835af9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2bf43c085544738dead013d28ad8478565f90207e4f79167a38d593c94f747f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "636dff46552e2a5b04748cb1720249c9171e8ea24af9a9459dc39f0dd0be3dfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e66d6d29dbd1e9707007fbce6af82bb542a574be2b24727b8c97fdf5c28d093c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f23f91acb023c275f5fc07a5cf8e8b09b3f141a241b5ddf2ec9e5b9abadd9fef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c43c2d47407d741c8c4d4a229c4050b8e6dea3c5c18cf588e511194c951261d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92c831fc3b680f1dc175230dfc0e719e8de92e715e2731bf0614dd238d64a0d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da7582d1328ae3f5738d1280acf0a2e8de150cc14c5c476e3c3e1dd7395d7ca0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "328e12104423134a29bcd9c8833770e29ecdbf8b2b9ff11246d2996465b82fed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7d11401365793afabbb23a0c0b6b93e6eb121019b1ed14eae4267f850ebbc172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6cfaffb6d8534bf2814e7b87904d6bb8ca8dc3524f15ac07d024d3978823dc31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43be135e843dd566616798f704ac523956f951606d929d71ff3786bf58b85a43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c1a460b96882f770af1947e7abae2c8512fa35892eb95b2cd3e12e115701aae1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ce6b6914dd9305fe4c9d98f3e24313fcbed2f61a4c6d4b0eabb775574c054c9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "104323590267384d69f049ebbeccde56987d602d4073d6d92ba27e98f4abc281", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "719c3e7a3dde1fa42fb68b5700641850b2de0d9d51e616509b54df133a2dd695", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "108905c2cbb595eb0eeaf39e5455a1e6a5992f5f14595903877bdfeab2773159", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5ba08afe0389c0a93b8cffae607fd3b9f2b0b8b943cfaed2d53913313c9a4b33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "90ff68b2ef879aeb2bf7e73766d4c332e7e8c48388dd38f354de54f829a5aa46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c23d7ca46a3e8c942f9c7c1f827a91994d448a175d9ed1f4f044149a297784c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8ab6cd84dac8a7338792ed24efcdfaa7b9a5aa11b05ff4abb6added1856793ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5a9ead2b04cc6001fe8331b15cb4fd4e4e69716a76637e3e06ed1423eedaae94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e495a277786d03c6c155d5734225e90c769e251be6d5f9832bd81c771b191e6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "45567e1e1f446a8fcf312f536f843f04a4bb796dfd799a19d04e2458c60687ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dea1cca595039f930611300e398e92ed1e2b19ef1d064b73a77409a8fe85a68a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51bf13e093c41068a86aa68dfe9c2dd1ea4cb360d2ae9288cbce65cf8dd7a1b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "608cb0f66ee17eedee47dff9b42c38bb36bb73c4899ae9de0b8b1e6ebbd7b46f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55cebec71143726812034df69780b6a21d24829f506bebf30b56f79fa8567774", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9e5295f82b195ba67a7c361d78ca64d264073bc735121cfd95371c212a95c9ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1f0df69ff86fa7a5b285abddaa8267694db420120a0dcbef682dae5419eacd4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7e0e7b8b7ee2a7113d20dab6f6dac4965210378d7eec8d8eff378a0943d3d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9b0acc47acb6546b429c22734698155b00a1e62c7f9d1de7ea26e568c9387c38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "01e1f54216f40111c5c130bc8e7fba9a4e09b373db1218b2308ccc825e89b75b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aeaac1dfbe514a9d1ad69b16ac84e30f390fbab47b84d0019e58ac15988d0425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4914a02d9f98b500f5adfc0b725320448331150686e6100e38b038e1ef9507d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80f3e7b1b8675d7bd1b0b231e6d9877f4e698fdfca437e7ec736a28bfccaed9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3243adb88a36c904fb70d5c4130427673fc658d53d13c343c79eed15fe7189ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "94477fa1a0f6a520cca9d1c5aae53563f1785d1853fa8f9bda0b6832e7d3765d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5d98e3b68feefc471bd245c7ade174d2be83ba3abbec5b39562fd0839f73d21f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2349970ea352a42d56b6531e42370f43bd052558879a93bef407719bf01378c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "196d75adaa434493229ce15e883be4476d52be8f50c557099958f4e24cce21f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "03a64179072380bd2c182e146064ce05ee615e4e34196c463ef890cf3f03c508", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5588c7d6f15ce46b5f93ff2e82b582e33178225820e788939f275ec5ea565a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a96b34492ae82da2f81f942acda9edcc9efd92ac5e67c25277ba1fe7d619888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5cfd17f1fa3671732d90f3204daa0b4997167c75fca4dbd76b617a5e01d27ed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bba43936d19f0fd20f2cea7b2ecf772d89a25e7547f62b3256c1e59e76785b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bf023d8029422a4068991ad944599c1b86fb25e95cf3590a7b292a3057d2d535", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d4a94f635c696537998845650520fc264a6208a351bd3e353f8cce2b0ba08da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a383c5d5905cb7a0ce8db4610c45a8b0fac4270f6aa4029e117c96e8744c9fda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7899b42e49fc0924f4de39eeba89786965f8a7edd4d4c1c465d1024dad40bc61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07f3b5198989198ef1344c18eb8b137edaf7a3a8c8045d0126d25cd6df3a6428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3b0b10ed9b88d5ed791aab6aecd127875f3dcc27e9313734c33aafa995102ea1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3599f2c4084593be5ddbedc8af0a62b129a1e2c8b0c2adb3045f763efc2c6c5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4d2e1043f068f2db04cd912abc89a2d56fe9442bfc16d23bfaef9e8ca9291731", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe5122a8e7db8e6c979d053545b38cebdd1efb22e1315b3da76c8ab6fa16d9c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dde05d6f04ee557be7654554698f3fcf3d4d63ba74b3f01ba0faa9fb09d8fbaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "762a752c6889a2c8c45221924b38a7ec72dc41acfc206a38f107e57c62f1aeeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ab71a988d25dd18532b11769162b6840f9ba5b66e8e77a7a927658f3be669776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b99f487d6b9c23d5b6c1234bae963d6df5fe14836837f8a27139322494f2ec48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "56483a50440585515154e54672a55318eef51583318cb71354a02a48be496ef7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18d48be7ccdc369e3b2ed9f99fac7cd396ac68c31cbef8706c3cee055b64a241", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a4a06997895d8209dcb756aaf745afe82d177fad42f6913b2a1a3f3cb00681a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4c1ea772aeccab33f8f2f060cc210d8df57ab22ffbe65b79743ac423b464e7db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "87cbf94e789134b83232074c71b270887c580ad0c4801a2d1a7492ab12f10317", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "2895f93eb810ac08c6a694c7741105e1ed5f010438f5ba13824eadbcad8439fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ae8659f217533e295a969b14c18380b5ad24c5af6aeb7b0cca648b6b36df4da6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, 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, "093328bbe494a342d754b30e2a5b821b3abe265d03903281a5f3d0fa87b40bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6dd4ead1ebfd918d71f7ade1b9fa310996c095c8b2c3be23209718081a1bbd06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3a323f003cb9f63ae7fefee1665e1a297ac04407ceffaa070fce17740688d908", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "8295f8c63030f398fc9d8e87ce412af4f089ee2568e6d3125d2e11865cfbe88f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "f1e8865500576eda09fc807c5c740c8d09cb892f7fd0effeb27d1b591d1bef28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "7cee0fb64ae08035c63c6a1b11669578cd06cbd350ecbdd46c45a1cf4b8dc3a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e249202529b6a15dcdf51cd3ffb160e01a3e45db2ddf3030037d369ee66a1ec8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "24239f06fc9d73218684e7bea54cb742979d6f27dc8418dffabc5fb74d202563", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "940d21d205e782e8699b8ddf2a5544b925d4a50424c1333f129868324b5c97fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "35dd7d93513bf1c3eaa707283f914bdef3dd30c0d21bafd878a23dfcbeb05333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "96a91a1b95f6c699455345b9dcdf6edf841f39395d87dc7260060e5c611cd6d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "cb9b621462f0ed1e20884a8cb5d558fc6d0669bc9af8ab7fae006a39834ae09c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "e6fd888c259ffa46407f4af575072db7fe912535bb8101419a08a8466f87476b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "522da2218f5ff9954bbc638f0dac5ab52b6b619938e93ad5e9b2411b5fff4bdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ff0b951d231024ed78413499de4de9c0bbebac215ba7cd09577bd1c1e5a83575", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "d4af93204f86d85f71dca9f9c5d0bd09b6ae79af108b40ffe0ccf1b2da9bb6bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "98c10e29ccf8668fca041ed6981e76323be59634a817570a9564324ee6b49ef1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "b6fb2f477d6d7845b95b51acd466fb6e0a4129e028e7b8b54ddcca3edf28a54e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "fd44bc9dc0d4431adf8f464d8fc6f1a455acf3e50105f6f31f8576436baad82f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "50645f796151cd4f21f52923e6cb7af6ddcfcf54a68ede58184113f356682585", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ae99f9c6434f52b12394526acdeec9d3634883ca8262ed9c2c2d6b8517fcd742", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "216dea2a7d5dc1835fa520b9ca3821c13bc9cd6e6e17a789ed44d2b88683ee42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0c87a5fa7d7a15cbf82044255033181857a4d9e84ac5e813125f40827dc5ed53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "29339a5ebde1da1a385928e11a02af75bc1a4d8106c4693296327c0d3afb2858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a175df135cd5a96d885bdbdaf158eb94cfe27661d201595023047db1f331f6e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "9d1385cfd27590c458499ba4b10f6d28f48a1589fc11b512c9deb2b161073d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a2963718dccdecd0dfe79d4c1290dc82ca0ce0708b82749bb547e7e373e8b254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "39ca0a0132c64e40b6e26b8f82bd7093af852358a6445e14e8c1518488c06b94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "00d24dcb85356f6524ac5180e11dd63e4fc1e5bcaa5261b56149c18b8aad880e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "0cd5380d119e6a2154f27b96483406aae842f6c4ed58df1240aecb4833db8c18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "dc2720538c658576f9d5345a17c413975a13140a4ce3eae92be706512ec45795", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "751f55f5468828a854033a92ef5a8baaa86c160b89210808715e6a6249f8857a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "af2af4134395eca97eeda17de9cae6230e84ce5bd784c4a4ed85134d357abf89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "37412729df3e2efb55d92b751f308146824db0bc93dcf49c75b0f9e4aa692e02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "295c03a5f5334c5f4f3490345a3f77d56f20f29949e1e1c0e771b03ffeccd83e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "3a00d0184ed0be05f633b62cea2a8a740035afe68844f665a8cfb5702637c7f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "cc53d2f9725a851abcfab44757254418e8c5580f1922d3c8bc4b624210ea40e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "60fcf595c3e95784d72c7a092ac03a0bc48fa12220e9d98dc232873cafef2ccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "6abfcf0b213e26623be85a61f784694a7a0d9e3bd130e3bb7ecce84541c277b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ddd18aaa84c21d75c76574a5bf7c2e5f3b978e45fd12e767b1386d3fe6c16085", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8a45f09a7f3576ea10639c766304797f4f65b1caf6f7c540816148d458c1113b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "8f10e9fd65cd3e819edd4c1cbc08690df1641be5e2be8d146d2b2979ac997d6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b0246161ddc2adf0552da786b91071604ab2ca58e1e150bc3599b856e5aa7e55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "36aeb3254baf68df6d8e4905c6cf79e965c0de21bebb03d4d4ff2544711d4aa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ea7dbe816e879e4c34285dea828691dadf6bf3ca631bb42696d0f0f6c6cbbcc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "2212980637b603a88c1e761e464f9c51b69f188cf5df0635e9727e1d4f07a087", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "2e6c6a1ad606885f4626ad580e42708eea6dac52f847e51bb93a28e6818b03da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "76d8ee3e6f9d25cbfb7af4e4bcb3d10ec85b5220498d7727053890f02d7f657e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9c98d34db9143f59f59a80c2770129cf97fdb56ebf430e9041e86ccfa76ad223", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ff1ddda12b63cb5aefa37b10018956d297650a0f674945ea62530a9a743f9b50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1cbb1e9b3576913be1e5aacdf7c76428f52ed0bb5d3cc0a2638af1721650d98c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "dc5970546f74d020e46fb75c7c054fc8f3216db168fd857e260acb32a8c42d70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "146695cfbd241a26976baa0fca0539bf179c376d702cfaa4d1f37770d17599ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1a99cfb098eeccab25f0f9b996ffce2b2dad0d52919c298c2a008e4706f9afc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f76baec64c40e28a1df06f4889dab57a3f10e37afad181d6f6e01dea25efd453", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "6494e71c27e65be3e3d017eb16d94cd57b3e56592ce04b46b5177a7c0fc15d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7e953c6ed22b04a54df69782db9d4c841f4be634d4323198ef1c6b7f4f79736d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "05ee4861c510cbfc3abc6a83a4037c7d07bbe2fbf0c818816b56f9b1b543a6be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f854dc0c9d0526fd44062f9906552e49c7dc9f39519f53ce565afc79fa70c21d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7396c24601f76fb5e5300caf79aa7a7e49c27cc92ade5cbc6faa50f41696cd85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0a1a81d682e31e7d1bd1af5cfd25c31fb5c1f7952908d4d49971f0fbcd048724", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4f3a7fde81050584e9f692517539f6ca088de4ea26f54f5b3a1be78864fb9e68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c73fca6a44713e923a3067cdeb08dd2b0f3f941b2fdacd97210f47d7b85fb2d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "13e478304a51b996fdf6b716fb976fad51f32086456ed764831922a58f5774da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "d180db45ae8bcdfeb23fc27493aac25f17d34fbd4fc87bc1a66741fad6112881", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7b9fe21636dd6502dfb59eec7413682fb73221e70ea9ea109e6be7f0b02af541", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "529e7515ef8b4fe2949e1f660c7b665179a27fc460fe3b26e1937c2b7e75c489", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f2a2f05dc995937cd609da79b2212f81f9819ac8a78a9db4daca5aff1380533f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "3c346d331d052ad48d3a19bd31f459bad1201f10d761939ca2c47c5311c84dc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "6eca7be2cc6c5c1b5639217db4cd9bc9d8e6477c7028481e9e1da2c89e48d66d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "f1c8800559766047f6b2aaf43b9666f44ba16e4a9185e3e29474d7d570436492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "f6a15900a3724114a074487881b6b2371f5f6e983f95d196c67a8a41a74dd767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "6e21cedb69caef92ab829cd0fc0adcab4bfbcea07c70a39ba61e34a11ef597bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "b0c6243e4261a62ec7568726c7a11a9e14ffe8207d4395c56b28808f8a85e162", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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_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, "f94ae82c1f9c551f00eef0eb2c5d8f5ed27eacaeb9c701326ea25eba158f57a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "268983a5522cd8cefa94339bd7f75a25379b9ef66d5ab2c35b260c645440f380", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "36e7145aa446cc6d8d43444a8cb5e4debfc8c6cd02dc1c14fca0cbdb10455744", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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, "5474071882148ef514b615587972b374779a98a12d8ad2c693ae2298d256fd1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "0b5ca0d001658bb76dad25b9ece1b428d7e8f2850881c2e9eea0f77d3bb3f191", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "29fdf5388087cce8b9dedfbc6c3551db27b99944dec45bc99e9601ce13e3210a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_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, "c4a2c423a0eed3c86753881351eb285114d984e4dee725059847c1d5fa1ba5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "9584cb4129bdc1a7041722222068b311d1501191f73efffaf4be771eda51e8bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "587bbe580496907c77c13beeefbba89f1054ad089554c6fabac1e526629811f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "a9e7591072a2235c31beab4612fb7145509c1b494f5647397b55a0701d9375e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "e76c36199402b8cd528c9b69462e19a24eddf1e43eabaaaa4e069dd8ba664d93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "99ed7bf22809841ea6b20dff51e473dd004ec7e1d397566386a938c303f89a8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "2f28384af3393f23c6c69fb98870d4664c6ebbd24d24390dd9e1f7c074d2b213", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "b7af74a976f52ef234b246cfee3cf713e22406284ec04e75fcb7959c223aeb9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "0ca265aec355f73d863d6d4ba79f3dcddda0fbf5064ee13aac3af5ea32d42bd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "94aa2b3e55fda15881cace6124ef5a6aaa7ddfded72db30d13e8e1dc89a277e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "f9424c7a92cb762d2e4a73ca9c4f9904ba36e999c63b17139baf2b008af65e58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "9ec4f2a51065594e4159014a3fdee23d0b326fa32c7cee221f317e3fd32b0f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "23e02690d7d765f168bc1e397d4a6eeca6f9e130db32c04b2153f8e8714cf5ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1e093e2c0f0b6b6f6abcce2c0e84218c88ccfa30fd39b7b79008e4ba901ed8a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "e3f2aa231a4a3550b27150d26f53079a92a18606baa46e1340d399239a29a381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "6005265920cea8f83369996ab1f5a18cfd63de3cda29c662156e129b4908d431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a8c2828e1336434b1341b23e837221eb4f5ad9c62efea1560cf22494ce805ec8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "423b8c702e470981d0f5c2aa47cb3c045efd188283a82dab79ed25232b573de1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "6993c34c3296ee5729b7548a89d0ba8f28965b8ad05c279ba596e1c87855b97b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "5e598283d6febcabad204ff6503d4bbd2af703eae4edbc4fe7e37554624a39b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "3ee85f5089c0df76ff8413bab09b1ba7c5081fe686aa4759ca64704bddba8684", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_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, "b474e2845a80a36ec84b62522aedfc3e41f24476ab4d2781e6fecf24010ef800", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "993af2852095d7829a9b8098e80f3347d172c853805a19c8299a44c4eb4cc57b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "f8e9e1dfdce5e9f5bbeadb6c3d535b7fda804588da0be0dd5c0377aa962dddee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "ea20ef2650b759797a88fdd55361a153490992117d6b3a2899fde0a5a087fc0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "e5b637f3ad2cac0dfae3d67a0d680638ccf66832a4478f06564156a7f9fe68ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "decfca6d8ef7614efef5a5d2e16649e6104d557ba511a33b7b8fc79816a5d28a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "0aa79cbbc781ba28eabc9cb63e2f4d33333f07b0a7baf2a28db18337d8f3f5ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9b67dc3c6ea8050b25a13f19703039d56e2988572247d5b700e3ec2f817def56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "8e86310c79dd902ca38f5cfe21dfd02f3e2b3c26f29ecc4f5ceb6d434a63fbe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "60ad77dec3d7edde202b78a04eb9bfd658180a9c00bf2215d0464635ff4694b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "fff45b0c20d38a1a901e1b981c281bc3a170d9891d245b99663597536c6440c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "4f3f2a495b7ef34b4035317883cb3995134c79f0ff17b27fd7772b0240d7e0aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "2d36d373e7a8a170c37ea1e6b8c866777308efe6073b8b91fee6fdeaeb83279c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "adfeb7148db726f715f067472077e0e64f28cbcbbefc07329573013f137856f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "a6fa8c35072e2827a1e4fbf0702338e852f20600da207c0ed97e56c51f57f06a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "0da8ace7d4cd18b65552523659e496c6b27f4abf33e4664067441a6552654e8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "3bd53654b7519ddb6d00d48fe2e279ef8207cb1f4b62af40149cf0a944d5328e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "e23fa58eb3a98f2270d10e4eaf61afcb707736f32a862e017a941cc254e42222", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 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, "5f9b952e751e08f85d964b78a383c0ed6b9e704d68286353f6752cb0ab6ff3a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "eaa112f762947092bb9e4d5d705be611fd307056b7dda196060aba899c0a5156", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "55962eeb7fe9264244ebc8c0becc701603935e045ddd51f02d0f8e293ce4f7b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "5485c73502f0c2f3bde55f0d660d88a43e77aab071dd4610636f7f35da0ebd95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "1dbdbcbd8d0b21e6ccc443635176158e3b90baf3370b022d74bb6e036dff5ec8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "3f4bbdfc96125cd4db7e2081a9499d6fb40771f271de39740aba51e73de08724", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "00723f4528ff64622867b7839bcabe99d79713c03b4607c7d0bebb978fd882d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "57e4937a059e3403031143d07e2fdbbbccb5b26bd0ee3746afe44caa72c42ffb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_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, "4ff493beaeb639e82b3154b4461c5503a12370b98d77cbe1dd59d84ec99473cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "96da2a9c0dc02d743d819d7f8793c0e16a96cbda7d65958c31775e6430624c0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "2d15f4c26756ef3fd45b31482ac75469c992e3894e6aaa69d61fd879d3d1b3b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "40d9331664a659212bb2e60bcddfd6f25928eba599574fcf86c5908b4b4cc8f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "9b231f113e445328b0789d0cffc33865fba7e78fcc056b4edd96185724eeab5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "9f90713b862e22fb4bc700eadc224f324b9057bdcf9425fd3dbfe64e8cf079b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "856a6c8f3bc85670d16ce60dea42838f18e2b229b8fd39caf8ea6fdabbc5b3ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a72f42f7cf2f3060e8f592197b5da37c3d5525dcdfe7f04bf0087101c60bac4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "4eae8719c09b2a650428bde390cabca3209e6dc82560872017b2dec2787050f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 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, "304dd95b00bb9a98cc587bcce9901e875e98ab92d07b6cf3eea782739890221f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "4616822726d0ae3e3c9cf662f09da4532c8530f34ddb4826d83456023229c540", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a9171e2907c5ac8787816942a4318816599d231a94dd1824eeb8aa7a34f6d4de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_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, "df3350a976018f2085be9c349929b3717c4cbfdeb53f348effe0eba2b197f534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "d871a781cb76d9132f5cc5ac4f7e00b740034fd83e252d75e79a8444bf1317e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "e267e784aee8688f4edada77fb48dffa2892cadd82a57ac04f173173e123012f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "d5399ba468bae00cd436ebbf53e4f8ba603f2d326635b12dcd5747fdd8b949eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "ecf93706599dd79d55761e7d845c4aa83486f87360187d43ae151c77947ea2e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5c19e1082610eb46c1d1028d3fd1eeb25d31b170a07a4a9882ecb466242b68d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "8737c07c505ec24fa925db9d763e84965a8362fd49a5ae5cd948ab6a9ca41306", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1230717d81f92d7ef4f1bdcb93da3ef105c0f120a648ee0ab0fc7518063fab82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "1b82dedeb5c296a3a14da12ba751302222fef4efb90b22f9f7e960b62cd97ab6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "1c399203be321b51aa5ba4ba474a2e173e7cc061eaefa0e879397a8281cac9a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5324d2eaf4b3f53e6fd26f671cdc2aecd1ca946c47e2207a4403b24621ffc78d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "0850122ee7477ddbe965df2607d37c7b21d52446c75b468b9f99e0feaae4e0bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "26a7fe83b12a4f4c15947211aca64395bba9a7ec5127ac839a3c96304ac609bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "97eec3e559877c356ea7feab3c9bea4325a2d815a4347bffacdc2b350957e3bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "478bb7eac331b3c3855ca05d4b1ffdadf3ca17820c56c8e46f03d43beeb6717b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "a8e345da31c802d5f8c43da8fd34fa7e8a2e1b328150f6a7201826af2eec06f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1528383a80071d94bd7e08519477ce9e6bf93221b48319be01e18b16b0c3a756", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "68994049782bf1fcedd5e6e775129fe671a82fd4c80ce9704b49f57499d5d9e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "ed73670639de60d7c49322e1e5f815d7852149b9b41e9a6692fa46345fd68e12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "bf00dd13875bf21939368ab92dd364a02efebdd329fed8be08fe7a96fdae57e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "729094712afa078403eb1fb4d5e677620391031fa3f8c68e82aebfffdd50e4fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "ff262edca118891a87ca1c0525b11376ea9ab344c743897237b0fea88d4722d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_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, "39582f213c6c344cf2e30a50ca680cff4eb1d1d8eb416a74253f9bd2145fb687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "535140bfe9ff60d764827f1073c164e8a8b6cc83537ec79ee149e2527b189ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "b84ef9a7ab1820295cc6390979b13b4b0cd538f36186ea49c4a8e1113bb2e0b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "be50dbd5c8b475ca4591ed3c0021922676da87b711755bee93e0fb173b53b8d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "273e75b9d04bfced65be855335db232ef89d9234005691f82090da739329a794", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "0ffbec753f495cb3ff26c42601b05682d97d43e4bf0ea2f8bbc662d06b618c5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f5de97fa6f3e7fb2fc159316ea657053e10d04721a89f29073d032d013124aa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "b51658c2acbe641a1fe432698317c9a5d49e091ec570b078e8de8b11ec302f03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 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, "b39631b3fa7737737acb218aa354e27b628086860b84573436b81be84394a1cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "f4cec531f1bc9d61646f345010ca037a6701b8d5dadf5298a9a4311c29ad4cec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "7e4778abe22f06ec35f8b226fe7ac22264ed1d798617e7aab47ef582700ca9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, "2dcaada47c9e0e74a6d8d6ba0c56b17cb00329c58f12fa1cf5bd1f8211a30fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "55fc8aa4bdc94134e17bb777b803b879f8dd6a313de51c098b212a462fe9c731", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "c9464c4e45aa82db64d1cc9324d929c5bf3979ea903788d527016d8a81c7b7d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b2eee88ab4b0c5b903a90abe7cf55849fa0e3c8eb631f8a2e321d8483435f699", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "91e3f77526e59339224dbba331488af540549a4404700923f43a5ba91600542f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "289b2967ee8c3cdd35996dd7fe6cb7220713d7c5181f01738d748e16896c5b96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bfae3b3fbd70e03ef7c2dcf25ab7634e4990ae8098f8b84f5b718e7cddc4b38b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "a651dacfaec534aeb058872ad409d55b9946e8be07d459a447447a793e222842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "03a463a9ec045c664fba6b99a418cce01b3d2944a946fbe50a13fcabccb25aee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e40d0578aec26dc5454c485f7a51f9512565f600bdc962fdbfca632ec85f6d25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "41b1ad682bb901e34d3aa48c0ac9907837ed5a7a092d24a3489d82263d4fa249", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b14dc56614ec3740ca0a8567b4b420573149572c1787f6ff7b66ffedeb5dbcd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f76b811dd6d68c3fb3c7a7ee9975d8fe01938e3c209e2ee6a408e06cb83997b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "31d752629813717fa9d59065e1b9adcc2141a21b4bbbb1c8396e316e60e91ff8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "e560324c566f6ced7637c01f083828070ad8629371e8cc0d950fb56bd814639d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 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, "e3b39c155c548e57fc87775b47d0fd4b6343711cf431bf5b34c4f1ffc89e493f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_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, "4ad993cdb0435b1b69600f0e83f08b8b7f7bf08827dd6cfb2675b6f67f3a2958", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c99aaa1fe82f59433cf0fd92218508a0bb7c167ba0424a070b0f485e3facd968", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d0cdd07898d440f4aef3d647831afb8a39d6da77cd66c10d44a9866359edbfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "e1dae7cc36a10b92e2fbebe5e411bb428b1731155189c16e2e9474b1725542db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "70efab94af41327fcb789f42c8b4fd52db6861ac1256e4c74dd03a11b9e05d7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18eccc608f0941f5c90f12a8f55353acb302f7974dd60c5a9a723dc8ac264169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27037f8bd14a52a6f0c3fdba1ee2d99878140b832703fd87b4c5b6fc54ff23cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1cf3ecab1b417ee9f866436ee3851eb0e734a37fc9b64e42fdb0e324fbfb7b56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a7696a7f5763e53caa7479e063c19da48fbc851800b08d74255909f2293f0d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "30d82ef8cbf2922acfa11c8f93a76671d66ed20b2bedb50af4e63f106c33af93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "bf4d0c651973256cae5fbaedebbdbb83bc2c655066877c97da51a4a74e92583a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* 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, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7987cc147200dae450d0a2844c5c17276985bfa26e1ccf430a0837d0315eab41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f46f5d6ddc30b600eee45ad69faaaf0c7ebf78beb7086ded79f68e9b650cd24c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0f6edc01de055d9fc0b7322369e54abf6d967a7f5f47bf3a6a1858362d42a8c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5b0075b070eff5b03eeb8e4122fa7ceb0a23c31299b7d9c8f051728a37fa4677", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a6209cf8e8040b464caaf87bf7f16b4c85b46f75302863b8f4eebfe117acf248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e7da91203a5b3efbcb9a93d263bd7b8bd5de79307fdc0d52fd1ef76d5deec0d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "24b0d91a7c6195a935463735e0acabc4e05e281641e04713d530e26610ef874c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b6509d24041e0d688e973821ebf7652311486c96155d339f1760575ec34da75b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6715f9e683e395fda587a71b2306ce9837d1c116893e6c5b27ee4acc60d7920d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f938d81717b3549db6b0ce21cca1432b6fd38eb916e54fcb6bbe9aed3c28034", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c6f2bbb39fa0dd1177dafc48d7fdd0973517b8450cb469bc0073bc12cdb73447", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5f65333c8d18db7cb549a70dbe7ca407bfba1065100a52e97a76ebc9c1d80e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "451b0be381ef4aaf515929d63313e04a39db58ebe268c5cfd36152cc74530a46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dda3fcb0582eff0001bcd3d96ba32cd1cb689ff5ad1ccef9220a953da11988d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "733e09c3527189dc6df795a3ce3d3d2de5e1ccb5a6e5292c49fe04bf14f0a465", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a7f42ebdc48355f5236fdc6a5663c3fdc4b99a4348be17456f5248ae5002cbf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90112702abaa637a68009b21a65f75893570810378280ba2173e88484a9ddde7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a3578d865022ed3f42d3ec21c9fd939323c2c8d1d7904de4b5c0e07ffba5cbb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "555dce3e2ef558d9af50ef83f6794c9a0a14a0e05e0511bf87de2b8d1d9263c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d7ff6ee2f4badb4a98e0308c22bd5530aeb59846392328c94d986cc8563c7a68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "18117412f022ce37a7a954ccb79b8d4d810cb485543b1e8d07aefdd535465c7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7f4711c173f6b7607eb28915922634a01788ba8ce6ae96b46730454e99b499fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "0675a7378e429d7ad5e2da25f71fc4f2b7c8c04ef68da86d9d7035a45cf43d1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "adfd5938c2246ab040962d3cf8b6e00c9ca4c59efcf66bc079ed00d5591d07b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "3e4d842fb3b46c5f9d2cd8b03f4697ce77b98503b172e85be2cd34a940022cb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "422fffb0043dbf72fae4fbf7d5c844a547d59bbcd4da649e56247a462ea183fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "66c9bdb234653992a385292b8c31e0db5c2932fb4fd0c337572644ed081ffabb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "a0021a2c10225f9b87acf67cc93e141ec56bd9f7f2dd26afc34d59fe5a79f03c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "192a69c1e1ff536c59339cf77ffe999b090b0637105f36e3c20fcaed422d93ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "1f3b6923f0d5893113a94ec64f288d24c9f36f182faf77bb848342f9e0508304", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2124edabe607846143b057620bfb860b22d5326d93d3c1adc34baffc805cd812", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "435c4ee50c90e806f05a23a430909aa5faa9b740710323444fa6b9da4c98aad5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "001816297cef98fb545956b48139b165a18208e5bc8c4f44062f3459f5c42092", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "63e423f0d22e24ac875097d699b535c4f87933c7723c0d5b707c55d5d3f2ae66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ab62e31ee96ea34b44792837c2d1655e52804302627712a70b64e4c9878c154e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82d5541dd441970b39dea5184063349bf433b98b978ca7dfd62bd0a1fb24c1e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4ba36fa1074735d2bd88ba0e37a7480541c3d877694ed1d7fe2accee0bc8acec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2d16c2c47b9628517e15de8722e14b1ace60c16ef8d32b2c176980033b9baed4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29df216dd5420a436c557bca6fbfb8e55a3495760fd3af7ef5f45cb1823dc160", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d0c8305f3bb5e892ea7c482843a0304a6464f510ca4c8b6fe7b1edbebe13b514", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b1703f1119fead399976519824fd9f05a078f927838678b3be5a9afbe50df9d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "12a7be10ffa2e9eb5cc453587cfffcc6f875848b016e5a99ef5fe0672448a646", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "12580dd434908c6408ae582ed1faed67714543a9db00fec7daeac60189aabc7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c84cf7a33069beffb460a37da09e9b4cb56deb4daac945b3f2c9dbb23cec0e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b3047cd94111e22cad536671e1587ed2f11db386715322d4ae8a477dc7e49f59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "280d8f028611e5b170e16b02c7a9a2c3330f1a8875576658726aabe880ec6dd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d7426513e863bda35197b5ebe74291a0b245bc6585267745bdd721fa9142a851", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "515c318a5fe643e7e8cf55ea301983fc3de1874841f9e25713520e3c781fe59c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "61db8da60682b0726e9d87345f4107589696913b733c39ba1806e5fcb53f5c19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2ed7b30c10baf00daa2bcf13830dbd7dbd2ff68081f4d66ad4a0c01356b8b48f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4ec0f15aef710641e2958ee3ff6bbdd39be3cea0677216428b0776a6aab4c825", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8598366554ea13576a3af8ba3d3354fd7e4a375e19e5ead14202c14713da3af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "65bea641387fc64d4a58be84bd615b3784876f8b4bb13f2f4c6e07390d7e6c6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80abd5f980156771c4c1797e44254c88e342c09ce99202d99a30e3969145b178", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "056b0b682d2c2e2228efed10dc13c9d27e984353a85fd83f3ecb236f96293128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f54b7b24c2974ec098f04f6f5a28afcd0cbb1fc186c07c7c803e3723c58426a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aa9f92e72a502c6f0fd58b6479310e9cef425a748a51484ecf216dad5976ad9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ee35969cbb560778b84c729961530803cf686eaadca4591c0d3c1a73bdc8513a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7691c464608272f53fe99be8e28575d01f30727336af8c28ca409cae9229d201", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5819db96ba362e1bab5d2c86ca088dec72403e5fbd94d04a54b180e75cb713b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c8b2ad38858c714e8b9f18f631752ab472462e6b33b83d79cf5d06b9ae20b8f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53269caad8b699282a04a2bd8ffa9b701634081d76a0c018ae454e532213b766", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "67bbae32c6139c659a8c629fa4299af5d3e7561f6fd462ae1876b4deba07c89f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8f50e3abf2c9dca0f407fd558939b4e49780fa63452435db20aaea39c1417365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fd3d76e0183db072caba6c16cd1bb72f9847d20929e9c09133f45616d548db86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f4e3e17430a603d1c2c42302b942ddf4ead9472acd328bf57eb07e5fe36f5d3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5dfb98a0b73119e2662afd4cd234ead2519ef49d1f75565629be9d4bfd3ee6ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13d60a0140d87ce7740456a913e015df946db2b51521e449db45cb8d074ab56b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9df6652339ad3bf9378a90be2fd2432f8eb256f730407b4f10eed46c31c4d3c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "66dd6da3dbfaffaf530d8231ebe03c4b6e7f0436c40f10bf609c54800f143d7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c367b0ffad17d0498510f6276882756f9eaec76cef123acd7fc55c5002ae64f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cbc9a7592a88c5a597deb60b74200f3102a2c42217cb7721cc892e6b45438c33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e5efbea0e720adb9eff7f16c9ffc62fe5f7b57dc86a7d8e1c9686a1c04887bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60addfddd546548b2e12bc673446c984eb35e073260c17f7d970497f8c328f0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dd9ed08e3a000586ed3db6f0161141d58b7470e2f2c526aaceeb94376527d099", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "eb788dff19e402f295ac25ecc8da6f167ac1396f3ae6c08a644b1d1a1738f79a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "85a8aea97ad5923a2baf698ac5d40e932fc9da117b3bf59b89abd9c3c79657ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "90ff11cd0d1f7c9faccaed7fc5b78f8a081289b9131f9d49212b54375fe76d38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, "4ede3c38896bb9539f36e4d61fac39c94fc89859c5349b540a4a4a18ae6dc22f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11c93db02bd7f7828afe970c18ea474f16dd6e9bad4d8fc05e096119414a1216", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20b01631b5a60a3f1b494a0056b67e9f56ac3a389dd0f420578664c7d954b6c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "325a9300affb30c36497e7f0e6fd8179cddde297c403e06e248cd62114c32d47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "416fa4deebeacc24eeaa70e0d569ae8f143643290ead0e11b9053c15384986d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "84062fef6dd481385b5e950f827c70833d6f9a36f66512bfc040d0977956156e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9fbfb7ca45839ea03b244ff03a0ac20c42d686bca7ddd7a527b46c95aba3c324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fde97132d73daae02c3af342e5522bfa36d7f89f83f2a427b0bff28710c5e009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b7388c07167985cf10136358f9d9f81e9bfa692ac215457de23940a2dde0e774", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "109fc5d02cfe72d030baf38e24764f6d1c48a36fe856f662d680d8f4e37eef11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fceabf79a242c76acce68cd351dbf994d11588df951b88d8999d8280c6fc8441", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "531e704a705a0ddd09ca4f02c06430436dd0fc8931e7fc241a4e2408510a29d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6a57abc8220233a1e745a368686f37fb9d1a08d1c6b6b023964f0e529ad982f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d661ae4632450185dd985512ced207ed1fad99ab2135db352eeea140cc71f74c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8a355ec5db404c27ba8dd7146f400fefb12fffea8d53b4c684cc47776c86300", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75b9479db43e974443e73ac572c498a75ef215fff88a1af331eb3726295361a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a32a4e4c20a93ba8c1a5d24c1e7329cf7010b40e23da43adf275f0b3e303eba7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6769850115b432fd1d37effdbb03df338b944c84e634a4f5ba7ca358eeed53a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "834875c4efbde3764eb7dfd2432c7c36a1bc1945659bbd703b7fec5a0d565b64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6b27c3392ea5cba754ac88af47b820de929efaf4dbd59c63201fb04c5d913d8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "69790a6562b5d7accc2694e5455e823cacf9d262522d470de3f3125dc3140f6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42c129831494d5bbc082032a92859e605cbd04115b5140b0210ba6ef509d684c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c88ea35ec0c428358c58c445365d115c63e6cdfda08794a6bd486528e7906d87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99e2eb9497b7e6c96d940b63c7994e0c9c243cad4b68b92bc6aedf373936f004", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2947e3b82fdd92a14bc66dbd0063d3222f790a293b9e3de5c61a9cfa75da8b85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4f0bda6f69713e233ee403d2140eb7723dcb76a34bf5c9ba3630163d9440a849", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b87dc0e9c2d4c24e4b814f84279039d4945fce17b615f0e0e63a11afea022343", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b4d0a9263e3fb152c9eb13894148d670075579bc99d4d4873b8533564654bb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ef632d2928c5309098d8f6b8994b20c577fb3f29444e655d6ecbce4d5c897e32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5150c1ef012c50d0518e2bbb8aa6eb8933977e5fe0fdf03198abde53f69b054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c82217224e221d3830d385eb0aff3cd5e9bd09637f4a7c3bf5364f0cf84b1bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "43f2d121b1256b06d333f88c47db0180b4e6178009b1e96dd05e3145b4c12c33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82ba7c286c04d7a7cd11a652c14e051aae00642000c8d45f832c646cc3ecf197", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d6aaa23a804ea854c584705a2aeed0a8462ee6382c261519233cd7a6d5bc783c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad359577be13076d0ba284cd78f63c672622ed2832b67105f72f894355a2eac3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8fe5d0a341d7d11d9a3baf347f4494522e15c929f60efa306e48dbce5d6636b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "590d40ec01a3222cee4abcb3f683dce54dc3f7b1c451d9953f2d08ef2dc8d37e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "1011e1a3a66f3eb76f0c1b0821f2beaac867225ebed6ea7d4e13002eb86e934a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "42b210892331152f506b48145c0bf400f7c5b92adc995ed5e217e8650bddf6d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "53ae7e2a4a69d722c5528ac72e090299d64fbc1c5fe38cc0358e280e947211dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "709787631824dbc17306657daebc4260d759491729f6140a390f23ab4c10eb87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2c32c8e88220d1fcd54504925fb75072cb7136092ac39c6d047b8d04670d5c94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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, "402dc1c771417a55783cc5751251a89993555a07af25007a8a521f8de2569a8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "af370d45157022408638ed39f653164a4199e6a237be376703eb1c9596d7ebbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a07c35374a0f2458c48d5bb6af8de6173c67c8f79c763a4e7d1a845228fc89f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "469c5572edaad82cd7b2e83cedda7dbca7839cd903c10419962b58cfdc03c548", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dbbe91908006e2c895575545cbad7d47715653946b2043cab85dd35a0c02f69c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cbcf5111c73238f2008aaf5e7a32fe1c5e1e1a9e9c8bfa4b968057ca0e640a03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b82cf967a001dea24cf67c0a760b75db28b9dad8a13bb9cb362656c292349c4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d279c6aba8474aab926a0cda32ac8f488f136ef9962482f9247ccd8d3a8a9d16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d871d5baaec37ba6c8e88161a07345994f7b4ab6fd2d0044a5c5a35c0dfde862", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "af647a9da67e103922ff2259946284bc54ca586a656583bdaaa4bb980bef19a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e299bd53e93bcf5fa8582727f13881c6deb69add61edc5545ca77bb7f1086885", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f5f52e3c575c3a15bbca802442f0e803e938db650369427ffa09614fe8b94594", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d4b327faa5094e099afc67ca41b86caba2aebbcb8d7bdeb131081f78aaf341fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8fab593c9b3acb879ebe6f83fb9f65cef2c7c3e56a540eea18ca19aaf1cd03e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "36f1223256ca539902bd37d4928c15ebd0950679b9105212601730505984ef4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d9a2a41c4009e5eed362ae0d93899595752d73c8d80542e922f12c139fdc41fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ac5f4c15c77d811d3e52ec2a9fdff69b18fb36d95d95e86bb23b8c678220673e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "932d3e55971f72a32abe806cb9359bc137587004307cbd2d4d22208573d48497", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51ef0b1a6e88d615da60fa75db079d3df43d6ee4bf0bf18ce6fdc48663d042c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1334f7a07ff50493b1ed291b58b105a18133d2093be99a266d72644004946080", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cc3c5eac6aa9afc2e13b0804eaa558009f366d1eb623b8df1381eedc18c0d80f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0d306cd11cd5b47803d9775949c29fdefc42c768fe64e97acad4bcd1e43d9715", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9bf0a5c812825131fa64a629cf4753cbdba3ce3f86ae219f72cb9f4ce9052eae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c6169cab222d6a40b5ca87c57bd3ac6d710942dbc6d43a70dd9d4ec1a730c8bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fec7848810e03714d8ae812dce5d23288b1c16b91cdbbff939f1219e591d123a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "53a12c49fae462501ca4f27521a74568eb25222f6bbea44344d2b6f3998ece7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "16caefaa5b976fc64f01001f727bc8c452e659def1c1ec18f4d7c25bc40608e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a1b294320e9c95d43e72128acf6d80aaca6260def1f1a8253b948792945cefc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14dd93787496abdb728d14133b6f4c68b6d85b07310d84184a07fe7f94e11fba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e31422abd34f7a23fc5efd04aad3110f96692c4a323689bda6bac5c479857315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "14b0870b366909c2564fd911933d31f5c2e8276bca558fda365f9df94e3da8e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5ce58bb18c66fad651d0d2bd09184dc0061a518aefd6343cf693c08376e886a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0335d25c93bb38dec7174b508a41978a3b59dc1c90f2319b2dc2207ac7f29437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1e69f217fb4b8233c7dec7a200a29035b4d85995ba5c6d6ffc10b684a79d6cee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3777282fa4371f0d519464b7d86ac88ada3d994994cd6c12871933f0efa4dfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "301ee31e71b7826472476cac89727c9b7fc5f91576d5cb1e87c3ef880e76d3fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "733ebb83b99605434492793d19a0d7c9d3b4f5b54f6d8cf52fdd3481a9d1dd64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d59e3321861f2d0cca030e5e9253595327caa56d25c7523b18d23510857458fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40bd38bfbcc6a9ac54c8b2976b490d13e6658b466be5d1963bf1f3b0ff1d102b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "89f154ceef25e4651c7ae55cdd176ddd4c61f8b66805bb0979174cc0d95fb903", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "38a506e055122b3fd767798d29217f1245d83838e08e23bd1e39ec30e272462b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cf9f0d5c1a2a685037f407d68157b03e1c3ce4c9a9b31849fd1a686640b56717", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "722907243d4841bbace90de5211a113f51c6b09ad6c743edc1037e480883b6af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dd1fb8c944d2b2129455880890469f878bcb73ecbcd24cc974408e8ab2e84829", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f4fdd60710216be2fa8ae4db6663846a26efd27b993010200cb7abe185ae7543", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3eca381f68e6d56267c44984cd5678e3f9afdc5ba2f1b17c97efe4db1140df5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3e5b80436df267c145d48e3552d02ce858f7d0b121844edcf0f8d0cc0bff0cdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f16fefeb6b75c1789dbeb7adf061d1a68eeed9bf6b121fac2be2de9d587c65a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cef6442f349d0c973d5c6e4f9666bff2cf4c0dd6c135e96eba0fbb913e6aa09e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cce33553551c235ae05880200eb82a9a053ebbfe78f81874fefa5de134f58a7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "17ccff86996dab43a2219a3b446123f73174641f3c410d496f3fa9a4142205c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fa494bdd4215e0abaaad6a893125b3194e548c861a7a37b6021811dfde1ea425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9c035a5e782629b3222348768a9b8ec84028f34b0c4f6eb6039fa632b20f7339", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4f2f2dd6459d83132bbaaef98745b2848f00ba30bbaae4f69c320a650e690963", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b95ad566597d547571ecd89514a820aacca978c44a018246e403603084a8efee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1ddb0d3e06add81e2d93249277f9f609534afb9ca3c5522d37ef4e2681a8b5f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cebfef8fbeab736b86e5516d56f64b95f75ba0af8ab1fcd714853c26897a06c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "96ee94ab3ef2dc52fc079485b143f52841a3cc69213065fa3e132c89d041e54d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "27647a0d28e878e1c90b2678e52cf6010dac62cdd2a8bca22bc9b5e228fb0d22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f27d95260b46575280c1e9072a57499c34a5952c75a452caf8dcfeed8f2c4675", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e8374136666161fde34e90a4df67a2f69037fe7e089691b04a402fb83e0c14ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "895bb5b132e29c9401798e7a2dd35a0f18ed4c7cbe1c43d5b9d3b92c36758011", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9592ece991e56f38d3e465db1180e04f27c4e6c5a91ad2d3f3d961b5c22bde1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42da777760389a38d0770c14fc6ba53e996493afd6da8a7db84b3e4179b54248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "59ec884c08a14db8d10353d5b0bad89628cbe9649b1c87052fd6c8c267e9538e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c21309d4d7868f025c1678f5a3be2dbe78ddb5e12dc3b26c700681fa0e8f7d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "59b7da4f52dec7a5f06748bd3f15b996e40ac1640a6410ab0c1cf5f9607ac076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6bb0daacc9353195e9a37e7a008b579e36121c3b520549bb5e6480513c17d389", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3f2e97c080e950bc56eca777c01fc29446603517a4ce41026b0233c7735c7ef6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f44be1df774acd4df47452fadade43e6dc1a9a31ce03048dbc538eb940d6b93f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8b87be30dfcab97d007829449babacbe995f3e1a4fe77ee5bc92f03d6467765a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "790a86bfbaa744e3961bf3c8ddf579a968b92ef5f5e4ac25cd8cae9ac8dbbc3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "66f9a21b32daddfb71f64285d06f8a8f006e6714ff9779636bec1559247b603f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "231e958fbc7f85d1681809a7430cfaf0fad6ae1de044e52e99d1763fbb754f97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a2f6048911ec1f011b4c81e308901c600e3e25aefb6d1bf63321fb6bb184e853", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1cdf6f53366578bb4604d9abade064945b2844ba4522546ec06b86a7c5ff2c79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a0b956a63c43c72b8493e7ab61b1fdebdfe0130780b3aa5c528bdce7f4e7ca8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a8f2e86927603de4cf08e9a282af929d936c15e600637be41b07d7d185275823", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "29617e425c2f13f7f837e78ce1e0ec98c2ce2ce70bcb6481df79c6dac8f434c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a58fd7e4845988a2de6de0d2736cfc11c4c6cc1d89cced823094fb262aa23dbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "169f0e88294169c3c4388d7a34ec1d3cba6e855f8e73333dcfa578fde420fe72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82bae5905c8f7693278a1b0588c6901415360315b1ac2e44785731d6060f9b12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "235f6d96f9b517f0fdae70154e7ff11ba8ce20902d0ed077531d01023dd5cfd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4d41913944cf0f9144c14fdd2ec06fabc8f3f179ec1b9c469361ed9d5e2650ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "266ef715a42ff246d925160439ac2ab5093aa7809c7ddc7be0801735bf5cdde7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "249739b9f0656a0f81bdb572787e6d4567285497e463aa65068bdffe8094b117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4ca72b0846c5538501d114c78ddbe99fb4a7a45fcc761eaa82c4364c7ceee376", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca2d6945c04e0c6bbef280c860ce2bc4e57c08adf9e1fd914770e02f6d3d21aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "13a01a9464d38ccb13ca5b1826f53798c0f93ada3e42dc02f19e001a3e613e44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51c3ea89b27d892aa0dcf70119c157cc3943206c6c58e07eb8f63060fe4e2671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1bfa2d20c3bbe0e0bb114f0f8b80c9eb596eeb1d253bddbcbfc3de71476dfd40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b0c87f06f8ae09055be8bd041ffbdd8e57a33637d5bb4856e974b02b16c2f72c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "74f05d1b5a3b542077510f3b57df5b5b3ce7325f512ca44fe18e6af0a1135fbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dfaa064bb58c120ab134d97fd850921c329c9d0e777e42ea8849a541400b625c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "30fc7f26d415493a9757d7adf96b9789538d3a15540e0b96b64b952463d365b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b0a7ed8310e92b7ff24f3d4cf44de8cda3a9da43a464d5195eebeee5183f0890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ec28d8b2fb4bc65c563749782b44f0ecbf2a1a3ca1b29060a39cc590adc09592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1cf8320617e4b6839f5a2f7472e6f025c0cbcbcd94549be73e2e3469077401a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8586ef2a895af024031fc03cde71ff99105d99d2971e7b1f1cc5cd7c46c5ed10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "83a3ca6e3fa799aa3b0145ed9e7c5525c14dadc5a380957808673ab8bc5a0b59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "39e01f01b58b74423306694da5d3a2c511b9d547e8c432ec05de05025516e9f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "04c4d17d84c199d5b78e09ecf29d84ee02a064da38df4771e97e8888d37df916", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1500937f05cd721c8ece47e2a4c4a0a27fd18096790713682ea93ac247e18fc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ad68491590b364e3bec87e9be669594ab33abab06d8702f5b140524ced080609", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d7117cab77ee8b7473755fddef42e5c126ad940accf384b12d580b0aa2d46e78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55bebd3f585726c8a9feda407532aca1fc39d47e0084ccf3e1a2e41a3d48a12c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a138103795f165077a120874ffc95adcc69b3626edcb88be655bb4f241a1442", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a00822f22359f39595e2b5e92c92481b49ee8658e1714e7040c1abb8c3976c7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "da72200a963c96b24453e7db6a57a608f4d191344c4e13af460f19c6ada891d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2aba6982d74434881057b2342eb7bf0e0166c3071819fde0e1f627b79725ef46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6cc0703d6093460fc00b95e01dc6086ecfed84e41689cc7e02c0c0a5376b0f23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "291a13f9446e29b4c1a9962bbcc58f574004cd4d1722931d75db05970701fc99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a0a3c5e226263ccda0b9483909ab703f3fdb16f44fa93f95850912dd9cf34f66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "21c1d7fa722fbd2fcd4b114e6df221a5181e8616f451fec8e586168b61c1c9be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0fa065c920b8640746a550dddff95e9883a5b5620c52f13acaadae169fdac36c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b598a521f620c1298a58b4e127e30492f4cc3c382f7f1212bbdf733bcba9a7cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11d0da73007317971eed6786850aaafa58b46b9f0d4a09aceeb8a7e4a4be4f9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "266fce2022d858400b50a6fd0b9b1fba20cc7bc69d76962d18af89488db60305", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f0fbdb2d936996ec1f693419cd7281af57aa1dffbb5385845af11d70146dcc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "cf2a85eef708700c4f52dda130dc1a60fae3c109230d4e0a82a5652b77d1f699", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4e09d488b2be781b35149fa86fcd8b612119bb9c6b628acc01b7fc49356e32bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_sm100f", 512, "0c4536f0c869dd60d55756f86f47770f7c083eb08c23feb650406aebd24c9aa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0edb5a6cb5845242fc0e0172c1ea6a4b68fc744efc3f535bc246a2a784a3bdc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "dfdb8dd42ce091a0fc25a418ff4c63522534fca22498258a3b2c9fe4dacbce88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7dd523a4c45b2de6b22ab395911f25ae615cf2c19401f8eb9b3026d8bf2d4fcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30d697d5677088a602f5873e5f9308d97892587ee301ba5bae91f8c9602cd5ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3a93a684ac20d50d73f85c3000ff0ba684840f08806e433437a40bf3f06c4be2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c4a310200ae7a3145f31af2cb71169d3b62563e99f46dd86befd345e64e40999", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e8130a7c302776ce9e9753d6c3837c07f99af64367db198459ab642cd9a6d7c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_rgTma_clmp_dynB_sm100f", 512, "c679cf6e63ee2f0c1973995e09d462351b135ce1c290e334bcb38149b69e6941", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "ba74977051582f3db6afbecf6b5d3704d6212ee71dc4a49cbda0d5abe35f1050", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "354282ef24259734feddeb2492197a0ef4fc621d3bc883800d87068389dad5e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0f416521c554046e042493202e1a014112a1c13efffb855ff271a9fb8c13f6d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "915a380e57bbce1b8ccc40e53e38d3b3cd47113df321affb6dd7890e019992e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ba50c969e74e3bb517643bfa9244995f3698ff640c789f35955bb4ede5f130e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0c90972dec2798d5307073c2999059c04207d0af5376567282f89c2467a57627", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2a1d9122eb7ba2b4d932115934c7d6f45b16e94098d12828c26c3d4716fd469b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "75b8bfe01a74edbf313c68e3f23a02b1312660d1ede816ca2dc7c10a74dbdd5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "31f9491565288e3b0ce7e5c046c85f3c5adafef1fdde7941bea8a63ec48d4aee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1e9171df5ab9a1bcf01efb31e2f3e36b38c38851518fe00c2dbb988346eae8a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dc986677f8b6e276f5ecc1d28b9d947657779c8cf8e2b6f991992bc76fb8f7a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2e88e569e7e4ecf6c71284aff97d8d840f71a1b4c446b2276ace855584150de5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "85b7b899e2dcd31c5b31a3b1cbe3e966567fff7267948265acac21cbbbfd1f9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10bbb46b42e1b8b865eb9e6a4f549aedb86ef6a0766cfa260a8c27da94f129bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a4c286239fc2cca5ac5c2b2cf635e863702709dd9af86c0e6dcb41ac9da923c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8a1dfbfac3324a76fd9e310ff32b88f4bb8462ee5d5521a883c675f3b7dcd382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1bc0f3b389a24d9b8dc124775a43ccb92e09913f03ba4ab4b6e6e0f17b791b2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0dea423da4d1b90f47496c33c1c2d65fef588596139bcc074e5db81e10f805db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "51164c1135a09b552ba62b5869146bc928adc1b4f966e802af79c878986cf7ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0fee6ef02d2793c49efb4ad8bf3333b63206a68ebf96a401b0171905750ef60a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "dd8106c90e835af93dd82f920ab4908dc992b3dcd8b859c4d73394ec4de62794", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "42a29e67134992ce9172364ec753de6a529a5122a429ffc3b37bb50d31a0329c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "b0a10a307242765938ff822df9be58a1ecaa2420fce253a88485d6ba3b955c86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9d641edea82a805c180b7f3cf373af822deeef0823a199b6acc63fd8726ad0f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b72a2f982dddb53bc9206a9c563296f84f56df196899d885718b6d8962af42a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6fe16b509464dcc28f0be81ed7af8d0c07e27ae67630c23bf19acfec9470681a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "177c9116ad9f47cd7e6e12d215bd766240dac140a14f42131ac3ec5c4105754c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_rgTma_clmp_dynB_sm100f", 384, "3c608f13c21869b3504f094cff2e34b54dd08c7691690cef42cdf4fd14780973", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "c8b5ee130b8789c6c996a2d1ac2b4698d0fc3b91b35ade97d15fda8e30e1a3ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9ad763d278965dc24e340c50d49f3deae707b667d3a1f68e9b23557f9f3c2804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "113d4305039d87010c2553e7434ecfdfb90c6d8927e8d33348a875b8e2d76a3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "54bc83ac589f3435916efa780508f19679e4911ea4a6ed37d65a3c70fcaa1dc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a4a986a00b021f352c6fb44d380c1931a3814ebea98f93665c17fdbd335d9477", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "78ae36a9b18a16c293f68f37304b2bc568bcbf123985207772c3fc7c14a28f1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b61b3840fb551f4cb359509edc74b2d81b114634cd4a50d63df1093a734ef390", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e43b7558bbc3495fd0dd67398a20aaa692ab950c4e357e9ef4cab44039cb906d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "af41d6439d5655229f812b92d22dd49afa73563f9570e3631dfd61b3b7225557", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "676a037b2847aba192bc7346ef491797a09fbb54d25cb307d8207b3b4ec8fc00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "019a861763ba4a9986c4c9a949f6c26fbdf71a33cf2dbb7aed61e8e12b9dc981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "296f75b71be9969e057a1f50951c660fd29575f2b8da2d16cfeb1a2d03b8ca4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a06962fb0f2196e2f9129cea08ff6ef644b42be31663b789eb5489a0faf3785e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a543ded3d50e86e25301fe9ac1a5b0bf0e3cdc9890f727bd3041bce8be85bf6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e0566405cf2b9bfad21299bb4f1af76d2120ac930cbec856c0b3538c17266b38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "ac37356814e82a8973b4ce4624d111919015ffe4f9ea4ee603536b6e47a0b1d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0968ba332cedd57e588b5fd24342f51aac283497f0c6fc91b62960cb895e78be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "e9baecec1036327baee139fc2ff58fda833447be346b098455a08a58638ed17f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "1d4094070a0cf2cf8f4b2f5c9882aaa80167d8156d18b8b41b8a2897cf1d4091", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9bc76573485b486ca1785d5bf058121ed7a27094f94b8c706c938b381225e83f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "b49c78ddcbfd0e7e4d6bc26f8104a13c7799e6e65961a52b4fc73359bda8925a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "7dc9a05f2e10c8a81cc4493f4150be92a751a103ea8a00a85293ce3f50a4a23b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "0099018708da761017fc08a4d142f00eb4e59ecfe562fef659a0bce8446ad82c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e61112f2f762f2088fdeedfc5c90704d79639ce79bd75b96bf35e8c850c1af92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "c4355b63649e578cbe4d34913d3c3df612c81813207b9169058a7a9ccabc8e49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "83d0ad512c2a2dab59616f276ed6a62faa8cbb75e488a83069c9231318cc9792", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ea31d5c7ce74343a27ebb7f65a859bbc8336994faf9d6de2397210737192da6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9f1c643f781128557cec0be5552a6903591fb291e4f98d6e2a88148e0c3ee578", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "9db68be02f982ea02fd410f88354d6658354a3fbe53219b39d3fdb3bacce21eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4210f4fbc22327eede6e208bcd206b9e22513b0e9ea6154969f326924e9ca752", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "304af2c53fb092d5c53dca0431e12f888cebab968dd46186c5e16c0a963ff979", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "63ffce2c09240ac265b3277f65487cc04a93d0a8787b4a59b0b7079972ab04f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e2c51d3e0ca79a113fcf16bd6e8d0abebe42df2dc5b69377f355e8f35dd209e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b3ea6d61a494942485db54a859c0d28076f7028eb14898b0efbaa4048e57c0c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f9c12a070d1a3d4a1d1ed15a75d803c362feb5522bc114244b046f6ea66d2a4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "8cf881f59990c18c26a0cc64667f37143d8aae48e6b730344628eb6454670298", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "0dd61c829828f12a46104f9b5f0eabefb6d98b3fdf12093f869fc07134d91c4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8dc40fbdf6d15be64469917a2771b05010e1598c11c1b9c022c40f0bc74552e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "194fed0f67d2ddc0de3b916b28befc1afe20b4bbb399a41ba8a547244ce0e44d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93c4fe47367edca16ce64627b9295154cf083959fe8f959017b489c3f798b8e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "519b95e16cfca18462303462dd03f4f57190172ed1bc1058f5882d8764ebeade", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "41fbbf1aefe5b64577d6d971fd109e5386f275ca73db1f6b9f6e16897464c376", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "54d4bbf7166b5fd77c3d4dfebb870a2f289bb30b04b7788ddbf7c3e41673e5ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5783861588c1edc0c52ea9f1ad88db45e7bb9f4aaaf0d92ce7daf032f7a1ec3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0bfef96ce5706a0e9446d2664ec2eb6dd5afe39b930570f217b0388cbbd4c138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "cc5952b58a44365f1b9339ca18169fa378c375459ac8381af7ed8bc97d1d559f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c4286298c92cec26313ba9514c8fcb7ac2f24c62d2beea5a72b97f1dccdc4193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bfca76f1c891246ced0552911b30a591374aba1a06f588dec625ad70ab2a423", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dfb487624b7747397c3ba5fc742198e1c35c8bb6bfda00f695c3ddb20b8b8cd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8741c6f00078b081c0e693a1b8dd31a4a098488c0c224ce06e8f19a8f8fcf2c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7a544e74b19f10e136aabb368d164640083769bd3f8201f6544a3277a1dc6bd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f53bd9b2c540c7e2bb4d2aee437b50c20ce2c190b3193b997d96df7a15dde186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f58d34bca5b30ec61179d802d0e3d58d3e67c004f5b0882eafd47e7906de018b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "848faea01f1662981ab4511fede2de5769f8cf12db03eeb8d84d57264a4c2c96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "54e4e91bd1abe6a5f564641a8e4cf55fa6369f55bc094283c31934a355f516e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99e8ee76da2189233c58806c89bd04755141008bc0d17c34bc9b94fc977c0090", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "006e96dbc40761dc2bb4840c9a9f2c4691fdb9c2b2a9e8b0d0aaf8f7092b54a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3a74957b9869a5e0f0581f3f11c6c0e07ab7b3faf465a84464d78af920f10740", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eade1e27e55460e7f4cf3a9710608baf204a8d3913a4ee5f670154d2d0d86e81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "11f6142ff1a0d18b5889adaaee72a9db24b357a4032084cccf8fad6fd79f32c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f40dd5c18815d17d7fcbfc56e9bcd88117e4a0f5af37421e1bdeff4b98e7e147", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4c1fa6936ac6012ddb641448df0bc9ee455cd65354eeb29b798a7c6534673217", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "773a95d09ad085fcddc9ffdc514388d808a31414bc248c7aff38b1a67f0e4357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ae78c29499bddc84f028258e5bd7a4a713f8792184c558b3ce6145790560c5cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d959f91de773c6c7fb506295d07e079e21554a414686f5589c02c361d098bd8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ed15f8fba1c778129c41285fc460da69e06fd86966197090d2050d0d4d2b6270", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9508ed1cba31ea2985187a0ef3b9af66c02b68ba9723453495d8c3bc1f920ed3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "316a3a4f0c68fe9ae5a30756f24e247b0b985dcc9b73ea7326cf159746790806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "20c9a742d13564fcd1fdf87e97bf72653db452d9c073df9e9de8f5568865263d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6c6773cb86759cef34368c7b5ef1e59a54c78b495fb3757cd7093a2585d67f8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d5fed420879d8fcb6e838b48238b100778180b6fa6e727cb5cdf9aa9aefcf471", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "47af172a31c11a94b62d84596e19fa34b43cce4b36ee1be060740ce1fba6aaa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e398e93cdf025b9a785185af8010e803c8d2c4ca32d783d56c4f8c48fc5e1b2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a7c998301574c2e220cad94956be5133220a0cc363ae61bc91dcf712ea91971", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "b14087ed6261b51480644ac295ad74c9a753c47c68ff598b1df054e0d6dc8e63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "d2a5c2b227f959a92a38d48a66ce93b8b74b737859248d040d784685ce111947", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "880b3fff3e5400ac6f378e5d3c2501e01920d42a96aff22932e990957b96d1fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4a1fa236999be13b43382b54c2f5ee31ae38ff99fd1556576e615e72dff448bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9b9edfb13ae84b89c522a2c1cd601ec13215cf843d0cba677dd365a41ab2e2f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "682ea66e900e229675c0b55067db43ba8ab898aaa05671290d58988b0fb80e7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "2ec79657797533cd7d6d371a1a7967d84c1f7564771da8b65277d1ff2d36f7fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a784df9194431df3c1c07aa36977fd714b6f91289e49f7027964480d6f457c45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "2b5cb32f9455d9525b4945861d9e83ff144f57051dfd88aac96d8ed8c8bd972c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "4239e82a8dfaa5dc58d3e12f43aa674338a0add416cb9fb8e7bf8f6041283a38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "12acc064b698b19c99ba07b0fcabe53b1284617882a910f4b6b45d60841bae4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "a0316bea04daec9e06cabd0f086ece677b5e5c2ca80954075fcf710e1d8669b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "681b182ca1e81037ca21cecd704f25cc9cca28c0e0ba1063c677a06fffcca8c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9e6587601bcea69bd82988c6debe3d6640f8407e344b3c8364488ba6561f4920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "9e16cfbd2045c1f56f4cc90bd36091262fd8e7fd77634141aa9e0cf1f129409b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "23620aa8daaf088337e05f496175fda97d4b8003b92992751efd451441b563b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9504bbf25d193059f16d03dc1ae92bdf52775e74f9f57219c08b91c15cef3d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9a0b7d2077d52e9414b4f3dadc99e29be14b2655f3fa2f3f8361caae4e0a55f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "caafe92e001106749d2a261e78e6bfc6f576381fc7af9b58f597133fdb6823ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a26f708db48047f60a8036c094f2cdcf170115a2bee067c00ac8fd1e103ac9f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "f727fa5b72030785921cac90b4236eb141e4ba7a971392873a38da5c2c52121b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a4993a6286b0e364925fdcc0927769b49550a41a8f2c1bb90a13ee151527012f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7696c2d7d87ae83e5c00a2e30c73e501214173940193d2be438edb6bd48b4423", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ab9b5112e70aa5ba3583841d08dd777e222fb0c99731c924a3740a5918d043e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "55ba5196fcffa27a59aeb5329aa9a97d142329833c6be99244fe6a5f804c6552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b90b5ecd92237745a02bf6f7e71c7623f6d158a6dfae65347e13700779867b9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87f54c3695232a4e27a7e12eaff5cee6ab5f3435889a31040b7327366cfb868c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a909b9b482f81fcf02018bd9330022ec6956013c63c573b260c5d273d7c540e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40f038c357d2d6dca72c626ff30cc1d6d32cece901c79e8e08e55be45f9e5e06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "65e211f168b7960ca63f4a09da163b00c6806c68a0e435ddeb4dc212673cbfa6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "b8b3a09e0ece8634e87113b9d8885ad931d3d1fa0524cc8e071e3407584e1a7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e1f1972ee0286a7e18a621bba359ff9a8e86a687be2d2dbed562620290ef9658", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c796d0631bd7354a2294abb68a79e6ac817bce14fda3b7dce58b90ced379c31c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0936d3a63b640746301ddd7be7c40fdb85f376d46c2d886370e39efd09eb4aae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "80b32214ca93417a604e8c12a9ac0ee4cf89a709cf535fd0dc2a169f22a01661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2f8d0549d1816a157cb2ab64682e36a0614ea781f471e4eeba136ad92a9069ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "04f22bf42fd71c11a8ee2bbef5ac401de0aa4fcf95947efdb693ecb042739fa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "87903a69740866262f5404604f0fb8d1d6920fe2b2efcce38b761d5df28c25e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c25a1ce440e893f1e529081e84f23eded453314af9ee7b931ff9ee97f4d1ceb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "db3400611153c755f2a524762ebc1ee8487265e02872856948ef10bb6a5b2dd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "aa9ff16974c32481b8c07eb59a42359e6fb26baaf7306ff47af0c7ee103a4bdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e63ce61068c812f2b756c1d83373e9dbef042d7986bcf7e2dd0b477f012ed70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0546199ad616705ff260fd0e9f0354f241be7049ef2576a6244bcb81e5f26122", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bdfc08aa5ed0558113940143a6b9cde4a75ee698200ed6926b9bf5f8270c5586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bf938600dc5d6761ff386ce6420de6c86b70d5335fb5f7e77e34d82a69de270d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ade38a70d4053ecac43cb6e56290196f4b1bf6c65f772a794214ea944c2a982c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "70d0cfccfe30442d7c952e6fb0db51b722f8c508588aa5a9bc4e3822b90d8d74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "68cf69be56f403ca618c3aef4ad2bdaacb2308195e71bbdddbdd3915e03d6269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f54159a6486d88711c9ed6e49f1b7f4520783b4c2a8d263900de79ea3c1cadd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "38e4c704f91530af77d8d96729d807fd9d7cd5b0150c8a76ea83a57a251f2b78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "291dfac0e8edf5b3e8b952cc681bd0b8e6e5ffa8ea21f56d56ebf900cf672278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1977c371d4f60c3904c84eeb52fa79ecc75d0a8c52b80af7c8b471a976493a7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "d57e95d349cdec7dc38e946045132fe0388e3888e964dd391f41afd8ad168692", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "74e4212c337024ceb1babec4d2b7e6ab1e6702f15d2ffddb74289a1de90e75d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "1ad80a48927c1841027a573896faf229a629456124b85bef1eff45a058917748", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "b17bbe8e883fe1f6160d7ce84e93422f95b6e3534900be428e497ff3949f4370", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6756c0f0f03bdf98b6462cff665e5d2f71e6d59d9a4c3b1e2bd6249f56dbd911", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "655ef4da630992fc0a57eddca272bab72f0ce00d9d06108056c9693355fc1057", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "1bb82b8a5c1235f0789f3221e347ab069e0b5b6439157912a6cf2707a5c40261", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "62a1e8db644fb7e013755f55c5320358736a03260df203eaf924b13489d21b7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ca14e7b718a7a88529623d25201893ee8d7cfec49454ddc0d8d6997eb6329e9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "5454a728340ecafaed7dd7cc716f165eb32ce297a444d07103482a4fd3c3e61b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "af9155fbb116320721d51970304e53c6b93a17346968bc6c9d2958aba8bece42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93b3a1f6d4f2165efc40fdf1f0640ce2a870a5d817ac78a1865bff606189201d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "5bf92eab50928b52b292d9c6a93b152b102a64261f22e6b669bec3cbe0e47f39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "32b57da0fc8ce5c755bc6ff8c541dc7faa395183478f24322a03a789f9f207b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "3d76f071b2643640171d21a991930e3de892b2213e5114495c3126d0a927d80c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a4cd61e3d7537ca5e5e0abadb96c418e4b26ab486f763a2d84c57958c82fdbd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "a57e5810b7685f52518862f43d937541054a25ef3b48e1c5bca789596d7f805e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "bf97c4ec293e73d5acd8dfdb33d9fa4823c030c38bb87281638bad88abd5e20b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f18510436d901d2a58ef43514885066ca12ce6c99a612ff369f7f567c6c23212", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "c7031e6b1fbdb3f4237e2675303fad6e51db6ea149ecbca2c2055fa7325461eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c31c3573d293988ef2435739e2c4f3b25616c0fb1734af8fdc1b44abdc1f0650", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "812010fd9e0714158918808a850b05ed905d0fa4033f945be575c509ac3c1502", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a64870e88327d9dc4ab599f3ebefefb1ad060bf4e976a9be5142a2a2066e7ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "3ebee8a3c8fd10cca6402eea0b30371e438cf816444629cccb606a85a7ee9865", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d7f8198f3fd6b453bbb32417ca2d0f36bbde26afb06c7d00a4827a9171e4f063", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9eb35eac82e9c345fdc166dca2a3511e4102548c62bc40158e98db337666eddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "261b851d261c15cd7f4a3dd833e7e7a391123eebc572c74ce8c44c493a913c9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0176c2705b10fc7bbdeca73e163adf70037ecd5c7055da88bc267908bd65a865", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "10dc26a4ba0177e00e1e2f083fef2ce38492de08d0880929532093ff50a2e18f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bbca4398e89a224e07a44f896eac28db7fa886af0cf1b885f6eeaf2cb5d62818", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "0ddc839dea51aada956bda116c350f84c80592504cdf3647d595ceb51d446bdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "793cd9e8690574f2d29b3253f4e7d5308a18ff5bcdfdffda370a51a352cee355", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e13b70a5a1fb142ef05c89f2b2ccf500238c3218e79c2e274fc0fd013efe6fe1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5eaf03058c52c2af4fff2443f5bfcdb32fa829c4a0a39df9d2df90ed3c3cabc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5fccccb0c433d38274619255e21faad83b53be427e1bae209d521edb324ab090", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a278795a16e795e9db86bdc873b41ad868ea6674fedc18c5b83e2bb44dd5f15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c134ead8eacc46d88fc8851dfc81ffee365ce6986254fdbc50fd0640dbf54ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9d8b2bec07f7e07a3287f37078d5b2b7f11863f669fe46eca3a849c0d64e4eab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "de85e512a76496e49fad8d40bb22e8b61cd1e5d1bab5fd1c94a9c303f5e377a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3b2cb648af9ee025dd8992b829f419c52143b7a240a9ab2a5e9a7f8ecc987aac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3b22dcb23c42afa04391c7b5e2e6cab36627a618035bec3227434833ecfca3ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1c45953271cceb67609c7f6f92d1358d123339fd172d7cb0eb4a7dcf1e4db3f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "211ac44576e821a398a1b47f4a70a6dfd83018d451fbc42952b8ce79ba2a6d50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d46361203cb0151a9b0475b5ca1c488fbb3f2f2b0dfe6d1f4fc9b374fae0fb8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5dc1028949b1932045c59f0df0cdc4ae6b0c7f373db0a215cce7a9455f9f6b0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "298801956557d7c99553a3e47e1823d49246d14055fc1909ff57742377cd9ade", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9c74c901d8a111e1134576ca5ec2ec15847de9c6424328f7720c5d6c065a4f6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "42c0cd5b98eb39d4004da0fd61c8cdaa05e80baaedb7e2f35c0ba2d5ee2fc000", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "92e79a37859197aab226b7b97fb40104e08a80142d8dc06bb0c3734ae6c40305", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "fba9f37a4cf8488159ef58e0da1c0ece174a5a44eca8adca1f436c5292c9d579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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, "2a0c05c7c4c1711f07abcda99004a0f66f9abb38e94575f8a78ec95af537fb48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "a971a95a2f7c505befcd65d05919ddf3a22eb41e8278a8004e7be329e6293026", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5d4b2f6ba81ef1c448874f4e59cf8e57801d4658e4274abf0d4720f1b7e1af9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "eccbe1bc94f646e21b54b2904d38e76838e02db78e7b7ed2857bbb6a80006da9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "001bdb5ee2945fca102d3cf4d87b8778790e486227a0808359c2fb8a416985ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "334e5fa4ca8de6077601ffdd47abceb19285680d223564edb013ce772e4babb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c00e7a099a01044e808bdf634d55870ae02ea9292faf1a4ce3b49d3ffc752b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fb14eebd7d50afa6034c40bb26d5e09f2ec1ce936e7554fa9968ae6ca780dfc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9127855e8397a4db4b5c004c0d88ef648e5877b34bdaa153f83c1ba0c1287e10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7163c04709588acf7711951307e312e83bc287710223392594c9ceeaebecb275", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "037ea9b691dff022d5592a9d38cc157556b7b8e33c58264daee2a91cb3fa7fd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "69ecc8cdf167a2bec3dec276975badafbd9f5242eba2e43fee956780e2ffcffd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "7fb07d7ce5f7b67ae63ffde66f9068d79e5029b8cde4bcf19e9033ff0875f565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5f11be519503e82c58ea4a96495efbef8c2ff458b4cafa4dc3dfe08858efa789", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "5031b66fa1a08bb28273ef3a9695d28aa626146d64c72b34f40846fd67bf031f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2e7f02130b77916448fb2bcbe3def92c680032b357fa1c5483792bdb1ed63650", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "acd89b18b67365f2b02a8bafc90fdba40af58e21e3eefad701e8f86cd30aed54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "998c63ed4a53d6cfdfac93fb1f19d5c3deb448469c074ebfe4c3af8ce13d9710", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3c291588755e5b8063173bad8a3b0e36cc949643b3ec88fe36117668634b9adf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6887866a1dc2caa9ce96ad9497b55b8943981a2663082b523265dd1d1139a533", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "df9ef549e412b84a2be9c738d3a189e0000454913dcb2cfa6c8918b13651f574", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4d64cafaa673f133985681aaec854c5a574aa76f5bed638444b901c42f0ef29b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3475a145828b7199168e9e8307cb5d8057eb3ad63275c3f8f70635f79cb51e19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e2888edfd35d72f2fa894f519c059f895c31e6ed4fdd966ea11fee12efb0700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d24b07f92c126871c3135c3cc5c447eb06b9ed101d9be660fc2e8bbbc507fee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9175f752cadf612ca9573b3f8dea90c70018b7713b62ed952c519ac1437f6344", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1daaa3f976c84621a9d550f8d2935601b37871ccf1ac4b25f757c2c12648d0a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99d7601a200a2b1a9357d5fb8acfff94ac044989c9f1d36601ef4387e19525bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e8a8987ce5c5780c62fefd23487a074db197453b099d69f45eb05af0eb02f65e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "99a7775ac89e4c4c3f13bce74c91e754f22efa2f50a465ef101a0bff4ab1da2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90f5a4a2ad0ae03fbeb94d805c5b895314230ef01b245c7f82a056f6223488df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4fa094c010fe826998fc4b02b8238edbde1e242e117f64f67c62186b5a056cfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1380bd72ebf0c413ca47e8b75e507eef1c877b50876c025dd3a21b564e285e55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "76ac4c57ee9eb5f651b31c87e3072fc634538eeab68c50f830761f91f971e8a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8c03547358bab441a3de09010671c4dc53e757947f568e010a841552aaed9c69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3ca89bc379c84560b9a820e2cba70139a55632c2863928bb10a419b7b94b80fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "efc5a9e0f12291bf6faba2604c336a1ce346e54a3bbd9ce41b6362b964e7fb1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "737525a56a1d4ae3d10dab787401826a2f65c0523bae4f566e3fbe238df854a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "84f85223d15f09edc5b72475021e2eb40b1b26fb4955d158414b430269ef3b83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60cd08d380cf393082bd8a53e294d65bd9f508ceceace31f8dcb0910615b28ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "40c5ed04d6579d36f88d601c2170752c28561495cb4976935fafd0c647619d9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c7f1bc6af674e5509e44085ac85328fea041121fe85a928526017af3a91d53c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f64fae3ad612de32b9190b34b128ccee7376006aeb36b6d9d5ff4c2cfa6a24b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "90dd4e46f962dd9707bddb6aa39cf10eec8e208f293f101f2391b1838baba065", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d840ca1e03545ca85fd456139643ea95477f308a7ec1cad0db6803903a9aacb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8e2e5bb3d5376121a490c3f3bc7c922cb6b8809fea6106b24d63de4b930efadc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "bcc0a106e522808dde72822964dad3b8237713b37c792846c387d5c27083d90b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f8fbe649f9dc96cd9b3341c5b787454af5c4fe61e8b121859712ce77551d4308", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4a4e9107149e3ddfd8de195469326b4328df1c151d6bef0a3ce851a3dd11c2d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d0f2c6b7a79868c279accc3d601bdf3c4e3fc3a49a55e10e9f57116f6a328a53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fe8ff7a4e1ee312094f925ed83f69de1fc6c7c8dd39918572a54969b50b62b14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f98a2880985f35ee7c038d35cb70b190c32059b67cde629639493ee1897bbd35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f586e3bac8fa310feb29d5036b4e6f1e42aed2982cefe1aba2779e862e9e1ed5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1d1d2d0252a24498ba05be850c3112b3a18365792d86317262a6dbd0679d4333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "514b89cff7148e3f2b61ff3dd23005588026ad4525cbe2dd3e275bb8b8085779", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6829ab6b40960c13d7ea37125e749d42dceeb23bdc4f9426edf95dec0b67bbfe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e1483379a664249469f65fac704713bab4667ac0f204cb2ba110c7473f505381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "e6e127fb6851935b466f71d7201eed705584e27bbaf617a0189dea2c3269a0a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "f234bc20a5bdacc952fb8e35381ad9951df431b920f21458be22ce74f5a18605", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3642f9105df77382992156d4a54ffb5ac7e31de2b4cd5cff3bd52c981b4621d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, "d3f8345201f5b1a70743fd5b821c3601225fbcdcb6f9cac2cdc1c683c25dedfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "88d204db34fba7d7a92e0bcd58f528cbcc5c71695a76d6d65b339ca7b132adbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "ed24b64d76b4ede5cfb5610f59e47819551bf92ece246683989322a1d0ddf683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "564f6e6a7a2d3c0da9890e1eec433ea765280625147c0aae46f514bf02628ada", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ce77347cfd974848fd69331bf4423ecab510b39d2d9809306104f825777d67a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "545f4dfc9a223bf3442441994898419dcb0fe3677d19dc3463245937f9d69098", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "18e3f4b1b765021a2a1c840d7c36ab1dbac55d58b0f5e5b700658b69c2546abc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "07d4dd729281146ece4a9f2ea0950938f97609bf590504555f131cf44c96580f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "cfd0f95a7736f5841c7513a48fdbbe78926eaef9f336592dfd429a71182f3a5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "82a92eec0309ee68b5cf76c90bc5afa924c2670ecfd3abc9413ffa7b5152712f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "953fa117ae296ccedc7f416e9163fccb8025b3fb14b67520c29f7dba58e2f029", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "a5ac0b55625ab5d9b8fee7013bf167b71827c359b816ffaf6a901a5aff0797cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "d0ffc7f603b89ace920e7e1acbaaf088b208f7539d73c60a6864c62dbf1c68e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "dbd7721725f0bc52a0597c2792e3799bea87664848f835ac760bb9bbcb48eab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "11d1acdb076c5a10068e6086e13d2e0c4beb7ba8c527f419a89863dc8e51a498", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d9d2c93037cda2a87a1e71299ce2d471dff480685ade6d142e4b1fd053ebd44a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4680b21df4ff8bad8b5c943c62b8410ec910f0937c2c6ba961cf4bac6995a776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "4bd608ba051928e310fcdc4e8c924a06856789a58e1f7e16d1e258f868e0c697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "45b2445799629c4232f27f4b35c569e2530d99c451a8ca42cf97f2494428d976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "fadd012b0741de75999f37995e6e15a32acb9fc10840d2c73a6a8f746c21702c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6d6f02b9b756c65ee7fd0116998628299a52204ed4f5a523aca78664fa720d2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "c3d7d6439f8ad4d015f1f383fcc67f289b86d295e2d81a3cb1c0b1912e08106f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "6443cc9ee8aa964502dc06770cd4c1c6d927ea079eb491c980b84e07ddb67075", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60ca08e71a4a44c5c8d92b59a0c41555c88710c48479b125f8c229efe6542c88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "26e5ffb2a19f59658f6e000f7284663b75a524f2c25c27319b276a341283ed4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "d09e23c1e3285cf2c618ce989f853a1632f13189461090e1b7ae1406a864e0b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "32898f6b5ffade115205de6ff8f1d917a46468985e15c8241ee82404a4b45e7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "93d8e23ff1d114c322cfde587c2cdb04405287430605f0c5d130c3355d952461", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "8efc7234221ba772092151847977325982e0df05c4f30354a8965273ebbecb5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "2905e0a6ad7bd363d10069583bfb1370a1a176af9c9aa169bab6affaf53b1c3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "e5f3a26f4c040e23d2002bfab8b5ae2599bfa2a5ac0253b2535f8b29dabba07b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "60ce7a63bba53b4044033b4d410170636f5083e4ffe18d6b9722e2d9fc24f3c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "448061505941c67ed52329fb93d5952d6f6841925c261775e622e155a93bb35e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "3330199e3f9ba3a022dab62c1942c8e373407eb0cd87a341d11e62ee0bb013e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "9a362ebfc092ca7be0ea32bf35cc80daef64d2806ee161c68443add452135792", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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, "1e68805d026da7569ceffbee55920770ab151b573d85c91d3dfcfbc3c14f8329", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 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_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "6ce33e124f9e95b78beebb3493565781c733f6932526ee4ec62ba00547f99f0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "79b7c0bb3ec42eb4636bc7548a1e873ee0c84725bd00445b7d16dfed6b7a6890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "f1215af3f963d505d12f2822437d96feea8e75075a876d83b5a4c5c0f687e532", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "d50c648b63f9691f68668781e2394cdba2b2c72f31f58520af2992a7c211cbec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "9e714ed55e2c11c4fadaa90d4dbf07ea1daca1c3e7494f90817b6a11c38e83d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "d21d9ee50f3d2530390604d8d3247ed3ae9693388d214509b7e213a3213f4982", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "4a75ffbe79da6a3a56ac64140871f97404aa727bc396de1f037de9160c51d816", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "62ea7c138a7cb3ea7aa3631d0e9d3935ff4e681f9ba758f21300feeff7155682", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "803bf51d9360219b9ab9f489fad0caa8a1bd49a1773fc001b0957ca2f7a10db4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "2df587d0148ddd76ee6d142d54c0863d900e62418ee9d703a2b6b78f662f85c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "8d6fd39b9db010ea5dbb49baf3da1dc45e03e6c9c6aa9be88eabcb80d7a920ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "3aa309e7b712a5c14e928bee2ace9cc8ca98b5159a4b57ba10b78c4e680b9e29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "7af19095b824bc8d7a207e0d2c1234b878b9c623edeb71d3b0d28a9b80a9d01f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "81e8f7c93182f9384a0f239cbba8fd20ce28a702a71b633f3e9511599ea43592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "3dd6f747e94b1a01f4393b6691bca3b46816f5e8f09ad2dd4ec7f0759b8aa9aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "69ae0ecc29ff8962421659a2884815eb7df1e0ad6cbed36336ef7a8811eb1f2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6bf333dfcbe84930ba26f011bc373f5b653f651acdec9cc1269e040dfe863736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "33a7b94cae16789b6f693e4a786c7bcae09f13b8605315debacbe3dc73806553", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "61e0df25a8585354850ff1f2ebc639a6755bbb9cc116db4211a600b1cfb2747f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "55af919509b5cba46605f1ce45ffbc43ffab6668a9ff8ab98d46b274a87c60c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "b007ea5553e3d1e1608c742d500bf8d9682a5ba55ee5d60ff0e64439abe498fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "265cae6ab4d296414d367774ace41be90b389c1f3ef2ef073af8983429f4a6dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "f947c25a6fb8ea726a91edcc27c35396c9ff9a74315e397bf0734a208e206dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "483bdcd90eda6248becc15407bff784994fae2ee1b10ae9bba5f8a834f9fb073", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "014367aba240456cc69b8ff9fc1af41af05db6bda24783c45a3442b74e7f2c71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "bc53ab57f4306aa57355f6cc4ead6b7dcbcb8a70d3deede590326768e1c60fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "8e9ff9210263e9405d47c468034135c74dc7163a76ac171b9287ad6fbaafcde0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "ddea1bbf94893e44404ecb468775ffdfb5c878851ce8287498ac62a9b6c44632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6ff7308cb2b5f6b3e44ede8ba1d22fba9b1b03a51f82921de1c801ec64e0927d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "c5eb662f695d6d54cb400111504cde614091eef1c8d8bdc3a1b47e72e828083d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6d35b263443105efe6efef353d3485ddf994423ca1d2be18d2f61874d25ba3fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "61e78f89e43ce15ed46974a963755c9ece2e5aa659a04688a90184fda2c9f9b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "5e983d5d699b1cf0bb1d27b1a8bbcbcb4f85b3d5778b9754ebcd0d1aa336283c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "c461bb43e55d8e50a9d77888543b9861a8f1e36cf238194d79ffb0595b4171f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "48f18044d85a43600cde8170882552b290aa1a72424a5b65324f307d869a3c25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "ea9d625576957fee6f17548068c98c6ebcd1873f0207d766eaa99d81f3b99173", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "1612e1fe89e0db4eb2ffa56ba3baf4547604ece8b7d672602c55b6e7e443cad9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "9926da0de5b5211f2a7d2adff0cb8657fbba93ce6818460d5153180d9f1d3a55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "68e7201c6b35223ab6fd4ce12fc26d1eea150445de3e46715a9d3a3d67f4f9d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "d17e2b762c7801428521f78686d25d944c5d2b3cca8ee5a25b688f2ceb29176b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "4b07c9f37ad3a4516d2775dcdcc2094be0a8b680c5deb3dbb9b977f576ae9f1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "e55acb0ad6b165c4195a8c8ffb9514f16e34e10abb08b93547d74f6c370cf839", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "002c638c8e2d2bd3b841b75361bd2038a1dfa3d1257b55105e42ab6d32f493f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, 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, "f843db6272365647a2e7671b3206aac07963533ff1050d6a224e3d4f26ed3bee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "85e6b9fd491b72cb97072c0c21d6046fbac17e73b4c063dd2fe3a1d6cf86ac87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "520d811450555e21f508b4305b3807308e1a788ec392b8271d6d8904eadc3507", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "49294b4602432f6bdc6730c8875043d039e6adc021868a726b136bbaedc62648", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "39defa0c45447b52200ec3184a51e9d3baab7d85dd1726df2705d16871db1f21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "f2eb1c3f70e52f63edbb6eabf96499be6066ddd3fd93baf7ce19944658ab344b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, 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, "6bd2a63375b65fd162aabf5072813c7e39271989cb954eb4befa56c6e0334b0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6bd4504a91962cec482c0f6b779c994b030e27221b975a6ce47e0bc976ba25ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "e3b8e88167bf378e861e74ac03c683ae1484935a94881d78b4dde8ce69ab4b3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "2e7ec31e5edc54e246a77f0b370fc16fe0ca5edc06bea48ba60fce6a55bd4a5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "3aa7551d75db7d1672d5abc99298864e54ccf2d0ce5755e18b122ad3b83414b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "4779b9465333398f347a01b5dec704db89554adb92ac696e023d940162bd05c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "0a366a82e455d6eab04f0708a33468aa25d29de551864f4fbf5608f08d419c49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "76d89d16e155a6de85a9f3db15e911d30fc8fb8e8fa62ec2fc081f673cc819d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "7ac5e729d28fe9ff7a52e3cbc79b929f16f1eae126c1ff46b59ad350f596b0f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "f39a4c8397370ea3837f2e0f597cb666c4b24528551cd649e62799424925fb1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "72673dad115aa2d099e7b7444be195a9541c73e133bffedd3da097a31d30d1ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "8c960ddab8dbf34721bad4d02ecd19580976a1df9fcda86c5255b44be2663576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "3e9c7e33507c18ec4830316ca62bdb7f7a1e611370afc1cd65ae3260310522e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "03b0a3312094771baf38a32d3696e06d96ceb5f1d9dd0c929186cfa31774e00c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "92956fb16a5bdbafc2d3f7b42cd81b3d34ac6bfb44b6f0afd8802392797f9800", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "19c6d9a8d86ba460d95ef4f93ba7a25fb92f0177c3ef9640584047b77893e942", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "6d9706fcda93951d01c1a54cb35fd89058d729ac8df2295dd3a3b070ea8ecb38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "57724a608b7558fd8a96f161ca389f97bf3a6051a5ef8528a5ae1322e2545787", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "0d032458fcf86fa06e81893fe38b1c58a8e2b27527142fc7efab960ca008e546", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "b32ee7f9955ab2563639ec4d8f2fef403b6d84365bc22f0d0188b1cab1f221f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "bc11bac3f4114f3e0c21088928c077e93eea00ed46b22d60ce9d07f51b49ec46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "fa5d573e3f5799a29cd1a378984ba37adeae7f1bf78a838ae83d7784100043bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "fdd8a469e7a75e2c036632cca51bd6ac36db73d49aa27ef35f293099c025b6b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "aafea4f8f01343a784d4aced827ca7621461c2abe65ca6686d2fdccca9ec5716", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "9cb018d28634d9ded3516550fdd7ab7c72b976a961fb04cf863310107ae3b9e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "a81abce178bddda395008988bcb299cea13e2acda3b7784f96e9fe7befd451a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "fa19a59b2c46f6bc795de1ae28883d33ec6c23af58e5035c18404dd7c81d49f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "52de99f3b67b90cfaa74fc9b65e7200e897c02f77ec847374eb258105998b359", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "234c71f44663279502625b7318078c59d9fb4773fbcc3aed237182e855730418", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "4f0e7e2be8dbefe8474f0371fdcf2132753c620e1b052c4668fe1b367b7d7bbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "5075074b94a78a9d9691bbfbcb874903aee7b4c7d4ed57af83412997d3d4d330", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "fcf23a8c83eb6bc2987f938f8c15b60a72860fb02e6015e780f3116216dc1e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "2c018a59ab2150d022d0ae2fb8db8d21111c54f8cc0fff9ebc39a7c0e4e86549", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "978af34bc0a59c1a4421f9e3593f220077307acd6d675febe96e8254c0ef36f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "276f8101579c91db3af7f2d63de9433ad9cf86f59f305780b46062b9200e9d3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "f0dca5a522896a726234ce0eacda052c24f30766d4f859883121f8a9247c9e96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "f1ea2f9364654bc5e308500de7244d1a61a8e64cdd325017d2492a429429eaa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "02e9614cf44f0f00673f24294e184ff1777e9b5b729f981ddb5767204514aba9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "2ea5ea253b7afdde7ab4cfd999b7ec6579ec665e0ac75c67e61a16d723b53bf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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, "808b5732ff0baa9eb74fee2f0eaab1879c3a639de29f264a03d1a382de16ca39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "e35e575327d16cad976169164bde9aadd4ecf0a2782c2595c92e8833202b0bd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* 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_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "636449985f41b605ff73447e8dd2162248f6f734ed606e2ff68e5d143fdfa3ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "839446e5098cf598dcd8729288958b7a1b60ee6e31986c15d7cfd34c4c3543f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "8fcb1c1d702c6199ca6f0dac059bda8bba4bf8f251c35466ca3e7a7a1654febb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "2658f4f4a5c4193e7329608fd64af80d3eceee8a7942ed69255e725adbda8481", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "c69d1039c6059b70b278f1c8d96f4f80f613c7319004655c93eef6000c8326d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "96401549ddfc45137b29629430866d98a877c7654c1aa749771d7c82be92aed3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "051e9fca181939a7f13e8feb4ec5de40a5e72cce73e085ba4ae13e22550ff837", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "a98be483fa0260caaa7d8e79eb77492f23a3bde7d9454e0b746e1bfde93838e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "6ff42224e3eb5071c80f24b39307268934d7c0f1eff7554e29421f6c4b96451a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "ba1491ef81951bc1912ee8a9419f5a850f0274bade18f8439534b268df8500cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "ba2ab8990e91912d09bb24fc4f97f6dc287e5f783434ba37508089e6079b8603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "2f8159d0389daa2b540b20cedf70802ff5f059857f91f38914cf1274bcbbe51d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "b3deed1de0154f4cfb9e2e6f34c492847f29b4ad1b6702d53d336736cd74489a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "a9f30727e835c1360d05acc5366a3baee264f9525c8c25204c2a4bf3ec710525", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "f2321cde233425b9f935d581feb221f4ad38ec1b22601484531f27e77d06de8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "552a1c9817a17722fe91cbf239e84a31f0ef2ac307663fcf7ebc3d7415b4c981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "9a037c952f859f34435a5b204d0187a7f31b9093eb7246f5a1a59928090c7611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "8d1b8aba709feaa131b331521eaaa717bfc7cda19779261d96d15e91f040deb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "764726e6f982a8519061f89794495fee623398209cef15020dfc2b77e6707e30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "532ecb7d72393b4798c391f7fa753826e6a1103b71f3997cb0736592b7341d98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "6849c818238aa915d5e355e24b69dbe918b68491159f246fd18b3d5154fa411a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "dc944e254b794c42ce451abd94ae4be4724e2a729af85d58576d174e5ad7cda7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "d613e2a2f160bf41091ef4c8d75521255fd945d37d6ea020926d9d76d7a1c4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "687347d09d05c10ced7309d07694d11059a80bf9991f339458d13f6f50e290a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "5dc32d9186459887f68242f57bd61df715893d40def08f27d89dd8a76eef12ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "ae74751013b0915542cf8f5bf5330078f5b33256a999c97b596a52227de48cba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "55e2f4b729898f69a2e003638d987c25a1399f43e588f15def7993234d6c44db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "882854bed2c3bd49c96f1114c20e80a770c12d55b6b2c625df75a122d4e6f6df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "ebb39a7b3d460cffa8fc78f690b467817f3bbc6a97c466feec1bfa73ff73eab4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "0b3804c4c427632ee42d5fd9afb53970ec230c332e5ed8189c643099f2eb0299", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "bc5b453024fb21d41bc152414c65477c1a50e9db797033e9bcd1b399981da14f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "e9f4e565d42cc1f02d210cd36ce4ea3e46abaf92aafaa943cc09e0249ba0d48e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "d423ac94bc2cad4f9bd1aa0dfb391d4761ec3800d6911dff45e76808218e51b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "be8ec13e64f14d1c764a7284e84a66725a98a4cab5756a59d8c5e831522c3e27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "b768f18491a64f9f955d90f13d9b7a755b303800887ef7b2258f88b2bdb3d589", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "8aef14982de51d7b14effc58873616d08110b736253aefdaf7579c6ec9953ac4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "87f6edcfab10a731b2bca684c13640345a3447b7d3e603c2e5332572b3c1c9df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "d8e7c8e9043dad9fb91eeaa37481f05deff1a4a6ca5936227ee5f773345b4723", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "0cf4a4cdd8eb56356080eb78db572e6d70f29cb4db5f1857d163501ff7141c48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "91f404508cff11087e6e2aa6391a9074475703f0c08b689d9b4962b3bf185dad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "8641701986af201def572742fec1e3b9ac7a92419f0c668bb17bdb6382a1238b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "075864f9800405f23e08156206c5ac08fddab625c450ffa308dbe4205afaedca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "0d34636ab3c71a06a8657e849027cdf8abfba20a65436d531f8fd07edec291bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "a2ee5053133be517c5e9228e732ec0e66741e271d4dfb133d3f4203c5f76faca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "00b353fce2c02fcb751492a86a7d173597b86d9b39ca7120348426876384e071", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "707cca9ace2ea3259147268b0b40dfd3771c82a4d44536245d910ca0518b1a20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "988038b1952855ebdaba9661c0e4b92782a461dac673b69223c5391324b55e9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "dde666fc0539dca3334e31103ad08642d4390fe2f4fbc25209aec9a6f3304747", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "2aa7f4b331e2d8354cabc9afeb0088e72381946b5143f2b33eea0237cb83fcef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "90d926dba0e9692430dd4a7ac2b081e680b1b2ca7c80c02223fefd96121f952d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "fc297f640a57fb8294587a16f72cb22222f365b71d3cd7309e2ea07dfabd378e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "32fbf780105ae58c6bd52b19bff260e1ba950d270e705b6e33c48da60c9f5adb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "4d0c5b48c5394d1ddcbea73b8e75562ff23e08256052f3c45ef39b06707b2c87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "95595066b8b77a2584f34144b919de48978b72942c54706f057b4041afdfdc41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6d47200d45c6c9aa9aefa894cc0fb318d870fbe95bf2e19e909602632e120193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bec29382887e6c380d827cf664db3aa9dd3093e274737aef2f12e03914d1f28b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bd97477f5f973d17bbfdfc681a732dbd727aac2b36b99ce72b187d90c4c4864c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6f684371fd915d645703e679cee0cc58a12952acd3e55cc43c57be950f7bb842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "20d1423033461a8c12258dd53fca924258ae56b8ee4f71170ab18b78354222df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1c206a60cabc7cfadf93b9bed8a36f4084b0e0f7a78ba45a5517c52605992977", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c6b494dda40652ed9be86a38f868c5662e9bbaf897be4adc294f0de6277901c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8131f9bb4d27398aa596b44b84bdc95e7d5c126ee01fe29478c060065a904f96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "717f8bb0dbe0bfb33b4ebb29a352838e29a6b0f24aff00c612da294f2dfabc67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dadbd624e467f8f6113f91f551cd355e2474d3e0ee3212329dc5a97da98ad582", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8a0076d20ca224fe1898f74a0dfb5379d50a575aa53468e0fd0a2a7734d0d7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2f74ea89a6a0b854aad833a5917df088be47f04bb6808a4bac5ce4336b5b5404", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b7ad81bf3b18fea9a5eec5d333bedc3b9d6c63beba590969b95a4d4da5516ac3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "de099fcef482c594219897abf7f2c2982b0ec91af846b99b8a4cad27cef4007d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "fb31a47872c465bb109bec6a0e0f9af1487f5cd1a490915fa793faf706fc1a99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "57600b1628e6a2a7d65ad2e4638aede2d3ba5c7d16dbe33d56f88a8d43a36610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dea02e27aaa8f8f93cc63bdeb514582cde9925d6ca19e99e71c2286e04cf2da3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "051642840e427af814fa9a21c1bbb2a5fdce80b4aade0fe7589315efa2342dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "816b8fee5759c406493b076f39c4873e35144429d29819f8b8e941d48c47b74f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2533a2a010ccdec3f0cb31289009bb4873f4887eea6fb984d70442dc5aa97bc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "08f78dab5c02c075893fa3b94aca442a5ab37c959a61acdc8ea132a9f1b411c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6237339ea389eb3be8b74d3c00c9bdf94459d60e75b72dd9ca64162eb36a5094", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "bff01f2c205e91e28ceea2b5bd53196b269d2936792c540ec40ef8ad8cbc641e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "59f5dfe3c453b8eb24d9d4cc86dedffe30b99577f3d4b18fe44a07c6c38dfd6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9dfed4362a9162cb0ee5af1f5743ab473e0451990baed55b2597ba52ee8f40ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "042ba46aa03eddc70e69e3edb6c3f1cae1e76e89bf2003c8057cd4a96bafa762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b1086431881d15bfbac2c7e2417db511ea23c6888cdcda2d5258ba5ce3f6092b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "641646fde5b11763157ca9d3896e121283d367956854fc71ea3d4946dc4bc12f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "370880c4dfcb54fcb1f58acd7225cbd14b5948ed8b98a83f987bf153ae603c1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ab623a1f3edbf502bbed970cfb3eda79f5fb514588ab5ec6fed75a6367d8a7ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d8aaa5f980fb835ba5c0101f9411b470c4b81b82df686c1f7e1669894e0daddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e22ed005ad379605986498e5fb247080cd888b7a0713d247f2a851d72e8400fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5a86d938358e912aef6bb04862c98f77eda8cba05a9ed85730178972e2383b4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7140319d35feb8cdde40c8ede194ea8f967c826369a424b420e720ae301d8e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2eee9b35bd8e2dc8d65daf535949c591a165c6210678dbc13085203f65824f98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "44124d5d6860727926f7ac86eb5d720f855d189b7bb5e293f675fcc8a6540382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9ea815ec2b070b963b9e9fed310a9e4d26671b1d97f93b2a5a42bb4f7cc180e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5d3050e194cb9ca475a9f4382cad5c1812ab90c15c9ff0950f93e933f0f3f5a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f7f01fcd745f3a5a2a0c818763d7af8697ee5f8d5a1de8eae0fad5754be40e1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bea534cc2713b03e7e381e9459b0739dec0e41dfef31ff2c459c7b4c5baa69a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b0f903fb86d62bd0303cb0a64138f47a56163bd580c72f129648bdca9e8ed2b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dfa9b0e06947144e17f52e5d29dbab9781df40079c8d6375e2a0e036a180a3ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02725f559a089575e819dc6af7b47b4f3e956f59e2064e6883718c3b2b9e9aac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d5a2f7a32a907ca9b9735f7aff43bbe547db32f9353835329efa37de7e61f9c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1ec88f322c1d6739b6d9d796f7052e3348fae85c082b9f4dc8c7084e7f7b7295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "810779fa8be4915c161f9f92b153c3e579b93870ef9f564261605e6723a235b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "37354efb3cef601cb5a88d8268899238f3198761d002660722874c0de6cb5247", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e9ddb132b808628f0592bff27a2e863a38bd180e30d89c159b4c10e29476cce3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "878961f0ffe6537c49224f88a94e62565ac10bbf9dfc79f912bbc133230bd25d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "59c9b789036707a6d98dd433594933c3d6bfa8982f2ddd2dbc396e702a1f2e09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "48172d322613e6fdc97325de8fc5cb0c30374c920dd7accad033d7f17070d958", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7643132d30fb7006a9f2313f70f6bd6d26e478d6a8c2f7929ce4e0d4e6713e8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "da3f93972331f5484db28a4fc785338ac0fd399f05e664491e66974d83d66f09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6a453c7fc203480f45cc4dbc5ca50a3591d737e7fd590676309c321422b047dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2dbf789c6c1b52f6693cf095314fc9f92a946b15c3bf9c6f85c43952e8b04390", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7d71f1da709e7bcb97feb4516736f5c040d4cc4c47316021238c8a39275e708e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5234e96a212adf908ad6ea1712b567fabbad142863412a4f050fa1f177a73258", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "10df1c612f8905e57227700a65bc05a76e0463bb680600def5a441f853c70182", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "657da9cdb869443690c4bdc90c329ee5506e42b8d53d61c41b609afb3a09b3bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "80d5a6528c9a3b9ca04b726909572442c20fc6e2dfe8eac4c113341e406b0381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fdada2079b601295c07d890be12ad3ce822bb91880f2e7b9ea8f0b4f98aad9b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "363edf7d926f7bb006148752ad5d6c4a71ed036be44ef4e4b6691e68f74545f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0defbaf8ce6276db0d06a90c47c63eee13727b5b77c7eedce2616fca4beb5669", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b5e7f46297752499fd1897ad961b3f5ae0ef47cce487a71afd2382f11166e2a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bb76018451f22a8f46a25d42ac073df93cf29bee9c4b1eebff0a3986ee91d7be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9bbe00376fc1063f1ffa779fd03c602dfd6cb3d3ded6c36ce415af060a3c345e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "41dc4dab99799eee66bb6dfdf8c5eb792b1317ab87cf4fea9139db12a3c4b98c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1c6d021d88110563d6a40061e9ec47529243681ec128f37fc98967460cfa9248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9bf8c8cfd445f60287706176034f9c6b9633ccbdefba69438bfdee227a94c9eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9c19d672e80d080147a7c8ef68952b94863a6aa1d300c8008efcd75d99c1ba1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c30e5314d9815b6a00a7b863ea3d39117985ec2f79b8a5820b840c0b2fcfed5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dd983d7223e732a8dc13f3568e14f15a56dc455f4077340f5f0360b591b24099", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "120466a1d73acad1f080e6d49060e9c5b3cea7caae14ffacaf07d9c6bb560468", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0600b581ebda107c46bbdc39a285ef316ebfa118983f8b9a9b6be55e48e1f536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8e78c2397bd9114e406006af6deb11e7ccf9d4001c9dc9895c5adecee9e3e1ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "60c9a7f3518747702085242b80222d461796d74e6a60918ed0d3cfd6fa0db16a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a95bfaca1cf19aae6bdfc2b422c70953ea0031f381b427176a3411a70189c66c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2c601d2528eea7a127182c93a1caa765afe177e38712aa9c33285725e09a7c40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7da08a96addd106db07bb5d26976c1e2ae8bf0a88d8f3682f3aa3edafd632895", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e45bda31adc8dfc57b3fb8136b5ff5a011159f35d5da6d509272b1f4e5dac0cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "df87f860b6cefafe797a784eeed91f858b8c4bd6abaa2ca17ad70e74a923d566", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9b15b747edaebe6d7f35c5cca110917e0255498e4ee14bcccf4e7779de443794", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8fbcb672d7184e70c1ccd152e81b59a8f9bf8cb781ca9b2de0003007538517ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "39ffae4249fa3ed54c3387a68e1f24f2b0b2dc3b8ac70bae7250c8ac514d32fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "22313a327d47846d68e55dc25f791608fdf9da6296444bb8d472ec1979d7a4c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1432a9a21071332390a57caa3d1ead4727e304364e67a46eb223c5076cd9710e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8b1f6d8f8bc137c9be19dd264a787118e95eb46334ffdddac3a65352dd5d4171", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e3b06a01722bf2ec5667c399af4e00c61cec849cf86d488895e67d0ea9a12f4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8e769036efcf042291054971a2135cb184fb5a17ebac303ecc64671a5ddc0b15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2191daec60e67081b4da0551902112916af25a0a3241f6bee05da7e24f7ed691", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "14df158c5ce218ebe270ca67bdd8d0ddb2035ca000bfd7fcc95e49f8b35c5438", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2bf924226948d3a0610b83acebede7ac43f7014abe19e7f8e5b25138e2b7ad3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "05ac0af1040a027c41861835bf57027db79a6e209177c040fda4be3452157dd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2ddef463023d76bf812cf72e50b38069a95d77a45e7165e5cdf54732b5ff1425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "24125b10148f1c680fc2e83223c0282e8b5ce0fa8e69320c41892480250a725d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "610fce8314967403f52a831b97402c6127f60cbbed0cad1bd140b1c8cab11461", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9c0314e4036276e7ca6f56e8cc9a9e0b0cbad9cc850c1ddceb8f936fc7580ed3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7a941b35b2993e3cdeff7ac5660d4773a8e32902c191d5170c8b1f8b8fd85521", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b1b31822d13c2029689f7b333367a13fb0dc5d0917743fc8e777b5c0e3dab2f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e77730fe27d918c3ce45b8dd67f834b6e61042354f2b76a42ad75883e325ce88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f86199efdcb3c102903f2adb497823adccf0bc654a92dfbc59206f8515b7c28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "03a50aea8673026d63d38893b48fd82e312d31b635e381dd9695aee5463d2d20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fad5f81d0a82d56f784bac47673bfd1c30b5a17566c39b2b83b1c937e55f7ddd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "e9111607277f92c67abdc814d4e5a29452c94a19f49dcd30cbcbe4d191a5abf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2adb7324fe54268f71a3ed3ec222bd1d6714c75383a4c5c1cfa6e1467d3f51be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "db9c7bea8c8e486f7b97b68e7305458555b55682cdfe3831070a3647e11f5bf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "db42010e5413e74467d72163bdd58cd38ad8c296c64a0a0b2f7fea862781208b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "be8beac1b66b21fb61d3fcbfe948d9cbfcf4a8cec972ff62fbbee5cbd61447b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "17a7edbbfce8bf6872c5fbefaf706bc7fb46dbd99b8103248e25b15486b6ac95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "45b73913703fbff2e7b965fe01ae7bfa127b2574c7cd556bb459d3e03872177f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a272c24ca15107ed94cef0fd30e89cbd8fc75838d73b0219386c05898cd66c81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4d03096395bf04d824653de5e3c3c1bfc6df675cb2769b0c39247ed6feda2002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7fe78c91dab018e5cf23e874540059197b02acffe467367256bba4769514e35d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dc25a59aa0dc2f86201024001084e9e76d0e434262b2b0d32e12f250db55aca5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7e51592e1c7313a7ed886ba3422be1be58453bf587549c5f5d59618b2d6769ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "851e15b0ec7a5ee7e16081a1d6b7a21f4017df4f05dbae0e294906a8ecadab90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "09d601513978a4799c2ae4054f168935144a9ab455f010f0bf3f6ebc471026e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4bf9ce2dff38adf2dbf202d5f8e6895dc81c4bfb66f11cee5545550c0fb7e84a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "5b24b72949592f9a0ae57f79c72f2d6833fc64b52bb3338bb9ff89bf9e6fef38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b08880117b8c557dbbf14ff4bd2eb9392b51a0bba64e8a0540df8064287618e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d22487c13b5f4a6e86e12dad16974a43814270807c18c03bddb90e93f531928e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2e9bb7c221cd2027b56a5424e7c6583331aeb4ee09af25bf730331aa2ea9faf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8f5d718b7431b836580c95a2b3a5599d88ed1951ad20ab1c474e098776ee1c75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b19ca87469e1a7f3cbb84ec32d42654b19a96375685cf37109e9d88b812c119a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1a1b3c1a2286e3c1a6cb658b6dffa7ba7e44cb84883e34b49e760d05a73adfbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ea02850c0f84c1e72a6feaf3e2ee4c2194d830f4d4d8d29546b4c530b1d07ad6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "314db7212abf90a931c207e731b1bb8ee8aace162a3a1b5d17052cbecec73d22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "df67941fee1c70afc9e13a91d66534aeb912048e3a5134f8a37532d43a4306dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "0a7032f73248291a8465eb8dce8390d2f389d7f32785984f82b61b5c1bf091b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c17223bcfbe71f8aa2515f1e5ad2db03d0a751de2322eace9cdc7e0daa997fdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "1967d9d6d55a4939ceeeff6392c7736e6a1d4b3dfbf0c5c25f4a8349c71e56bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8123c34539e034f63d982cbfbeb0f6df9e586fa69f633d25b47a5b796f7f5a82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "6c779c8a16add203261dff1889bdcb88cb130e0ad2bf401ec8677e0f8505c90a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "68b1b630ba910f4cfe668fba721788d7d9c883808ec4204aef27175cb986b8a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4ceb6019e6bbcff6e110b5586703693754a0e1668d79158d2dd731f5f114be18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "aecb02b367a4b6db07cf2b2d5c8fa79762de7e61ddeca59f69fef8a95100fe8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "f0a5e0b510e4532cda1503d523bdb1419b5eba150b375670ea8f5b5d6629648b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "17911f46235153fdd30a9ab039b9a815f665000f4ef11c458bfdded16f8c5ddb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8727865f920c9ca482eb2fbec67b4213665f5ccc9b2773b388df44c6d33a3bfe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "29b272c599b16281d520ba989d67f05a34807b11ccd58f12adb9aaa07a3f093b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "9c470dddd28e75544e1efbef5102da451c225bd65ef36ed4cff897d09e1c4bbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 512, "cfe9daa82cf6b951ebfde88989563510859743ad63f4dd4b66009e06d3325b56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "263e56f2d7917eb2e9bc3b69b9a214f93df597c7fb809fd93033c427a9494dee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "cd42171799570f04ffe43398ef8b51674cda7f46e5eccd20cd6570e4706b5c73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "865e24696fcb1b48dcdbc97570620facb8a08fe118c602ae847df08add9f26b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "1fd02106b11ca78629f1a302f924d881f0fd945a251e92ed2d9bb12ab0d42919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c730e32c34378f718f71889661ebe892b9b171bf98dd86d898842bbda353e196", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "33821f04bed8d733833733f1bc02284e3f33e9a0124dd6cadbd0527c703ad20e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "28422f492a62620cbae808c49961a9fbdc83181efdd49e6b0b81a390a50b2de3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5f5200f001e60203622ec2fc4cf04571676d441d3d4fd6c770b94bb4861e19d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 512, "4251dea2d0c3b1ca58779d8c7bcd01d63033f93a3d2e2dfad2a7b46bdec662f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d1cfa795974561678509c939572eb68504baed23a3cc75c4e22b15b02caf014a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "98e0565498148e3467a97df4f395911c99982dab218a52169f1702d5e4c9aa9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "fa64324dbf90f37381776db2fe6218079a07d7e0138e2ba2dcd15aabb56f271e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "9521b977b83ac631e1cdf23162c0a77652626d47f455c022cf31de9f580db685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "2b314c4071db19c4ee819f0bc8dea469ab4fab006077ddb2835d46de813e9bae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "dc300a2a10ce72451e377c3e6a1e78c4dd98a35f3b987cd407d1ed095854619b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8b75ca4fc74c92bb24cd9b233ac25434760722a307deb04aedb16f39eba4648a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ed2641217727d012c2853aeb57a155a88c2e2e861a029d11b3158e3441e34cbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "21f974fa60f2213c0b7f9210c93d1db21070930683b15d8413ece55f0a42bdf5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "0fefdb9d54d735b53d7a129b6674d0492b6d7e9044faa60b7346a4c2ac69688b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "466b39c75449f370e25a15491061c7c47206bc7dfc4a7f9142bdeb470c8378b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "289804a47d33b73dd4062bbe08a953554a09d78ea9fe34800e22e70d264861ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "906ee4f8e86dcb8d2501577466b362f3929951230153a2d3c5aa861626b09b1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "5458e7c8a1ca56ab29456c3327432d418b49d73e57e18d02f099a38f9f5b789a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bc072295d6f196d3050a8669dcea4d0400679651267894f4c0b300e8e28089f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "959776dd7201a040be30731c233e02e0973151860461b8abe4c8e9eec35b275a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2d8b33ba04b3d9c3910e6b666e4072f3178a1c746607a96fde07d775681619b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "243af6170e8be0dfeb3ea87edfb8f7a39bb16c98901763e1a68ab47133258eeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9538e6b2d362522a85a4f9c8301c23250f477f9bacee1001f83b988e710a8920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6aa202178a3ef1957aefb77f8523d17f24bf07ece6a5f8513c255f210813e037", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f19d74e95b52c0d3c3b97efc3cc548c6f5f79de26e818a378ff4ae4dd86d76a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3c5e20adcc833724119535f78cd867fa48501813b892c762ca75648301c7025c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "703946751680146ae6f33819848b49e560cbbaf9d15dd097c31b7a836fd2a4d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "76a24fbeac465d51d5c9f78303b65942c898dd9536b2f3215cd9f7bdbece9ae5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6e841bdb24800200b63ff6b8c5a795ecf7830fb227bab6dc3d8e28c3cf78bc0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e5fa0a6373937d359b47bd7d760baa3bfb1abcbb68ba4e97c5c4297625884acc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "7881825f85daf16144e38f649267f475a2c4f8ea498f7050525b78a7a3c3280b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d81edabfeb23ee500fae910cfdfb101899dd7694e14d694a81f043944f726156", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ffbc52b041942c91bebdf10285df20af75b68240d748240ffcfcae44038d5cf5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c92ddfc34637ff85c9f65d4e593874989a4efa6221768a9e600be3b734a2accc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "32a5bac628465647d456be2edfce7d82d224f5133f6956f515871a835cd9d9e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7848b461d84d89fece46a28ef8d956096981f26140a9ce9436810eb4e6805a6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "eecadf4ebae5768f4a608653ea04184a397763620dafd2025445e05351fff725", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2e828065ae2e9ceec551bda5c75b4bea973bf79122b2b2ef4fbfc26eef8e0d1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7828d5301cb70396f7bbcae150aaf54e02a52b6d43c8035babee118aec4dd38f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3005b5af9ffbf5c16d1f9cb08f5e5965336e48e2b59375f00114310dcca3f3bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c67e265b5ee2cb689f8d8e014486228e144838dd7f2e37b835f3750a49b861ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "f0d9c84c8f9ee6fb3acb96520fd7b3c46150b02f55999ee00965c908fda74ab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0bdeea94b124776678798fbc04c662bdbd2386a5de2983e513b38c4d2512bb10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6c7432aae908d781e71b4a53ca03135d113daf8a8529cdc770828f73116acb06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "786db10b191ea49142305e079b891d71f736d19d3674a525762ff1bb743744b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "665e29236a381769edf0020938d0bcb5cf4ddc8bc01338d3cfdd0ba67cbb63ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "10036f73a84fa8146b0f9e8ad4b2d72559c03b1ef5529df501838f574203b18d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "eaa36cbe821d786115a65b6dbcba2b0f94d9a5924fe1a94031500b1f7c8385f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0ba02b0fe7df01a4b943705e977f8dc1c80f47731fc85a5b9b213548437f56c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "ab167d325ef08829ab5bcdd87206812435fc23303d0385462ced39c8587e1ffc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a6c41012501cd7814d22370e9605bebddc435ea470053c9761f5663631127c96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "eccd6079de9911c6288a8f17a9f79088475cac7d24aaeb907b7134e4c53d6634", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "abaa75a94ad08400690feb8986f2b7de676e452df9679f105e315e523456f356", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "f247b68568a9b6eadb50e03e1041559ab76fff99c95cc7eb944d0f65b165ce08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b2eddaa0f3a1c94221b66e4e66c08b72eff55d878bd0191852070884f08ab4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "efe15e2fe8f087e82a91085bd56a3b011dcd7416803d9467ce989dbbc18324df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d8750fd91793ff172f5f453e82d551115117aa70df28b7a04d2d82096567973e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a721aab644019fece30ec522733cba6dfecfadc4103bc90dd4574ed8a913f2ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1dd4fd098d15d3db40146b9d44be15ec4eb46a65ef2ee88f1e161e2ad7af8626", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "678c62886615e8882af38a8ec61787c996f8cd3092bc74d03b37bab6999fb93a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f28ff276ca5e0c354fa0578973651e7e1023d4c8a66336cf1b11972cd5bc6d5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "af4cc493d57355a65d341a31b41079947d318c66e9ef78bc2b45f64273fd63e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2a95b8ca9c3017b59305284efa5460a1b5a35197e3536de3fa7a3511ebea2c6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "18f9b779c9dce0456e955fb6cd34f2d51733c3e63fe75bdb5cb920c67bccfdfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7a12ee306016bd67790035469dfadd3963264fd96ff349bded10160f79487b0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3e8649994c811b53053054113f91d48113fe2843681bde6dccbb9e6d222e1dfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c00048e7e013e6f480034a524d84664c6da71249b419697ac7d97bf05325c8b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "6688bdaeb9778ec9700e08a22d6a7d07b4866fd2e129e070fa10fd793ce05957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2e3eb921fea0bfa4f42100f3e48aa7ad4b986eb9d03c7038462b20ae2b2e2d38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "209ad0f743b681e48b09640c3704a6434deedd79ca2e42ba07e01694a2aad2af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a69fe08052c03a9797074d240a2a79e80771978671f0dc33b56b19a02e7c0a8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "738a1a1d1dba9d1749d85a351c9f46a3f2ce17bf76fbbe88b1b3957f765a4e1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c5470f6bfc394cbb22e71b15d59466c4122a281f0131f69d23bf028d15c2a45b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b8574eddbbb1119ede7f7d24ed19aa3f99280da40b1822e1341f36e16b84c708", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "174fba894bd6c27670ea4dcbb8b580de75beb96837a3fc11a24f0a8080550bc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6fd2254fce85326dca5589c127aa39672febc95a3ea7a8cd19ec9ebfdd8572a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6c0ec218a124ff4f2a24e78e7d852c68cfa2623bc0acc9d820d65fb2e53c6f90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "58956523bef864af956ae3ddb22525b544ee4e41912f43f27d7ca335012e3cd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f64b0dbb865469afbc427c4e0b1273c0815313fd7ea42c0425306087028553d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "88dcb789dc64049513329d29dac24497955f9b6d786b828a1a8eff5a3226e41d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fb372ec866e283d844d6220fdf10ea076fe7dc3658d8eec6dcf3ea5f8d675504", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d5337a5edca3315a5151b7202c794c84978570ac9d06e84b6f8d5c440c4c7aaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "00a63f8fa52c151d438ec1e699c00c63950ac9c5f5edf8f259b372b12e449cc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "edff1d975e22a7da864c37257ba04cb4c9a3c388182cdc0758ab01d789bc0ef1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "25d9094070acc3033afe74383ce12fb2be5c78a28b92d5f8a29f143213feede9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "826b0c9f82f249bd5deaf3b07f8bf6bd0392ea01e9840119b7d8cc1441867aec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9e04caeebf2cf5d76aa374a05e650310042acd4999c4734ee7d39649a91b27a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9acdfbd86bba8c7f18ba4af336f75bbbd133ad42b66aabb071655907f42ea4c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "65277e46d3b8df325d10e337f8c02a61e026b26baef8a86d315d113a25f36250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "4737392a5ec19945ac38f9a5a3a883f6ecff5cb3809e6d69f331b54fefda6ed1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "93f895600c5b91078fe30e0e57f975e2d67661af2d0e7b850072b97e65b147ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "80f9dd836af1c2f8e4203bcb70c6854aec400255e6752a50540b7cf033cc67f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a9f70d0d23a6c16c8f31a8a9b60a58efc7913624d452130e9c3a2404b7e347b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "62b699270deb369253d7dcad8870da751d1fb2cf69ee8cbafc08df1988593edc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f7b35e65fb0ef735334af118de3c3f8eac0976f4696e81e806712a68d3e3c354", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fb5f9447830f002fd07bc23feba1cb8614e7debcbe8b5367f71431110ae29d6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9a8288c1af4ad42f6331202cfc13ed5f9250c808272a4265bce34ace40ec6d02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "59e25366525a82b614d501ba9bce6a9774cc34289cad8288391861a3d804434e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "64f8c087defc462344bb4d09ffcb9fde3973f76804da4bf3ea7481492835c661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ab4076a5daa00f9a1cb3f4598cce80e05ff9f8320bb931dad73d2a95b88e2782", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "317c21320b6c65f4434aea9d3c34439865cf4b66d39518b7a9aaceb5bc066796", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bb5a3a8a86bba8b3909b5034bcaf11e12f1d9241e8d35c4ecff0972f3d2f43c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bbc9b865d58eec7946774fc81280279a398ce2fda0ad5e7f23831c57533a1192", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "3c6e9bdb49c1934620ed1a1f912a50d477ee0c37c7210f5cb3530dbd3f51cca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c7b29fd001aa90bc813e0170692bcf4fe9a4f8a57623f580b785ed105f484631", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "99f02dfd411b427fcb31c619f2bf14cf42d7b01d10c5e9a9b0de64ba21bafca3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bc6741d8596b102c09c29d6a0601bc679aaa17d0075772db4f236786a15c4c2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c7038254935a59734dd0a98f6c8235ee10bbbb9d367bf9c7b6a67c73ffc30ee6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "bf307dc59a5b4157dfd1070754a451e252db5c4768f40ff8389540cdcd673464", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ea2dab640a946a7fe7bb46b896077d176f30ed274c281c01c7ce69a33ebcb730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "610cfdac03f2fe6582dbcce8476741eb36d8d740bcb3971abe619ae836fdab0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f7ec1df0a209ac0cf0bf48c61ca77d544aba168a6f16a73af66829fd17ed459d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ca16b07a28b1eb48664bf72ddfe5da0c04309075b6a2f9a0659ddff62ac118e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a7a44a14dab5aa66e798b8e2d31d07ef0c9558506c704ed520cab98dc2f9751c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "de63b25bb85cd9775fb54468e08e629724dfd5449b0e19077489705af7bf7cb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "56f4f7139feb517a0c6c25c6ae477a2d35548fe14b60744de58486aa74d83ed2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "daf41e39e5c8e53d3e03c439fd05389b51a1bf0e8403cf08bf7fc1839ece0e3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "460cf5c8c72507c4c1b9767a0f12258689fc67d0b6194db2482c216c7e542d94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "944820ba28ecc73923892e094662e9732ae026b99fbd2e92996017b9ba32d1c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "66816e6e533dc8aa67da84e3855b3086c44748b601d56c8460bbe519eccd2d5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b56611a9576ec1a673d9d71f5414a8ca10db4b78c63f4def3849c56d8b5c9b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "39c900f5ac37eb731d4d4a5ef512afddc031ab0aa996e623d751db6285c085ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "15a1b3066bba21ad4cfda2f57421d95fc8d7ce21ed95661c319bcfaf1016256c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f4e6daf5a40410e7418d22ec5c3a77d242d5ac3ea0ff7f476c1d890997cded9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "54bc2bebd69ccbc6ddc1af35dd6bb6a69ee88600a6acd9ba2c6874c3b7e88d69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0207c7adb68e9f266718130a356b0deb255c96323dc41cb4008f346e7413d272", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "376724c9c1afd40ea98542cb822fbb26efea167fa2aab42f301e0414c55430ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c68b2b8a19aa2a058412fb1d555ee43a07a3acf61b9af512f20836fbc5c7259c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "061062cae142be495292af91c9c2f15c9175fcb087979dca003ae4ba6614419b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "908a2baa3e7be7c106d57aeb7e742ee5490072a9a54d20802eac485176af0722", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "cbc31418d1707111db02fd0930a86d832a44c230dce68f58621c12557760d00f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1602f3fa6090f35ea8bdc448523013e5d9617c0667ebd962cea6908b26bced15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "efcccfef51f4ea590aff67a52a997850096d9e726860c3c40a7389e9f8b790e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "45e2e16e33370f83712155f6b12427795371294a5be8b93114b0a44022b11932", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2e79e71a9b5abd2e11223a1bc7c56fc038febae1cd340c5111b287d9474f03d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "cd212f5b178789a4b09c35286f37444ad4bc07ac0866f9c27ecf41901e33f267", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "217e0aa8193af371806b07e97735c6610a5d05bcd50cc5b7013f86f990d27eae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "e7e072381652b8c200ded9799afaff1009507394e7fce757bfb1866a674affa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "58802af8df2dffce1b89bfb18962923976b066fff0d3a0541744809fbf9891ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "741954e1230824133bc9d6feb84d9cd9aedf5d41fe62662ab82b657953b74119", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "36f2b85be7d9cf578118a27a71a60093e07dca2a3dd84c2be115082b44d47715", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "e21693059ab831f54d52b6d28d300160625ec2a5472a62d5d96ae0ff304face3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ecba8cf0b72d5b6929186543689e1b18853c40515436aa3974a9ab038da92cb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0d8240968a82ab6818abaf8eb2e04c7a441c8805743fda269c073e1d45e3b10d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a6abaca069447affc8e17953c85e3f7ded29780857334338f3569a9f847b746e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ca0aa0e586b67c11c625c8a92c706b5a0a31d8c128290e310d7097574838fadf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "042756154daf77954c0e7183667aa828f67ee6a11be6d0dbe53c1088b4bdddb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1408221e08bc05e504c3a4a175437e8df3422468352eafd083472715243150f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "cd793c5ac5fc2244de7e032f6717044b9cf059c8cf3411e085c70a5df567e128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "47345ef937feb5bd43e2ba5f50bc98a064c492162b1562df787db2dccb26759f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "315c2f17f6c7f4fa9933ca530454d1ca4877d97ddd1ef28ae6716dd3833e4dfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "29aa9d3f04a5e90b2aee79965dff4660bfea181c7b53894e8fcfa36530fce41d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "990b158257d49c504d116fa7025b888b1b0bd13ee128a7a861d91a49aceda054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ed68f96de43f944625aea6156c84fccfd0a77f4e2ec37980bab40c1f04f1ec9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "03418b209ab0f28f6c4508b9b04c9c5a4f169a6c425289ee4136229d5d64b2c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1884a9835bef9a3237a266e6c4b5c4d54c5a1640ee726dbf17f8bbf676304374", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e3fdbc56a779de95560aad577172e4534dd4451e72f298c092744966c8450198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "43ac55a6fba3e4df475721af4c86dd351ecc4f8ecc463c7d5fe5c599988f7f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "078e0936726868e434635ae01308eff4269406d3305495257f381b5689652321", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "7fcc86794dad47f47e88971690a199aa60b3329cd306a4db95c7848cdb2644e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "561d49479a2814029a97a3ec5e766f30ee13f072b834d150f15d14096c2d4726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bff42e3bb4518dde3d89fbbf739afd382e96e650efb487e203239282e249a7ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "88eb79f23c13b8de1a4fffa1399847bf91f3d18fd5d14d704071a10087c8825d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "645d9d82393a10d677505eafd67d41354392e0d0e555d04d968d1d178cb4de74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "54b4348702197c955740e69bb523f380ff48ccad13647087880473816c4fcb42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f68e5b9dbc13a45be94ea7e7ce43b4a038363d99e4749cff678368ff3041334f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a75b33bb9927c0acf11deb5814553af53143d1bb22f5a015dfa7e1c8b02ab314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1e51bc7ff0c6c5fe6b5b246e9077054b4d0093a1fd597f4c1ce3855719ff21da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7728d2a328587bb907f7c55ccedae1c9c2223e3a3668bbf976d5ad13774f8e4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cfe3012f5ebfe4108ce88db15c13388a6f609765eb40756810ff58871bf98e91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "675278174311d59d37091ae1a5260bc867a3bf703b229235f3364a9a9bc2b9ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "05bddf508050db69f36abab3cbe8f077bec613ad6b3a72190386d1a8619e1c85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "87f4a2d7380a23cf0dc9b4da1a40f767fac15126dbacb35ea258e2649282fc1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8e1aac61b06ec3b6721a66d4c65d8c8c914bbfe363ff403eac1491b9f52cf4eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "8f3ec7337e5b5d8e708af3fa96ce694b954b718deacd33de8b86a03b42810272", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "89e6b71451c0491146932bfea33bd739d03c6640dce8b7e4514f126d0c90b4ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "eafbfacf212bc6c8fc45b6a0ddd6803fac2f865c05a3f89668b21d3594245e25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7f767facaa097d1967421746696093c576149dceead93612daa606a407b98aee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "af83532cfd2b2950493d49cf4cb6956f6e712354da9353d09d2cf3c9186afb48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3467bd443f2ea05287f8a4847d263dd114c4c0bf556c8f77d07422151764e4a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c2be3023db55985fd3bb39fd44b88f6ff31c8ea5c018cc726d0066534416f3f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "df019dbbdf50a558a30ec312c153ad82b679a28f9f7af295f7ccd137ebfa124a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3796633ef9391edadc2f48c5cd654b852dcb0024ecba3149a47c89653f52c54f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7b889f14965d65d8a416c82008c57adcc90f8437b9d85b9d359b2ef16a0b2017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0e132360d000a49f117177fb03444ee4c72228e175d7904f0e18b773808cefef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "56ad55dcda2cbc87e67fe4e5f760ed14227dd5c8df8d4c29a963c9c8317618b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "e67bad837e90ccbe9a4e6a9ad153a43b746c99eb5f4249e69a61608070f438a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "69e431bbc0d4a71702fcad6a8b37e9a4704500e8a7e8f6690af88fbd903c8d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6d960efc966b05c03fbf0dbb53c75a2647143ff3f2b1c691388d4090c30778ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "068b1dd9003ca19dbf5cc10b987bc831ee0075b06daa533237dcc2ece7e47f34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "64e52033dfb9e45241dfbc9287bc4eda69048131ac8c7066ec49078330a96c4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "630aff28b3705be17379e087a02f326623682a836cf09b7b84a6d0582e709bff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "43fc86f0d490ccd6533b3960e68d8312ad8786ee48959902f4fe8f01b449b382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4356da0a2e38e00bea3955dce0183bbc47c0551155e2b40184c20bf1c3e01d0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "61f455ca3d8f97ff1fb3dafb6035a076784f6330a748583f080e81966d3b9661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "06f894eabaebf10e1217401956d52148c2c93bd9fb56af70228059db8fac3707", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f08ef63ee1a20caa01abcbb66290f9375b2bc56403b6f1157dc3dbe3a7d70ce1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "71127b049fee11226e1240cba96aca82be7efc1c0ad90916f690bc44dbdbcd57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "93d518309fa984daf45e27793cd3a694437b5863b1d8df0041757ede96932073", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2b3ff62f009faf48b433deae1bb0c7684c76c61860e4f4dfa60a9039a55e0ce1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf44562a0f5ac05906173eceedeb1d6bbd34276d51749e8abb2c76fc3b0388c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "88c4565a47b623ab3b33fe5cf6a571cdfe42a9999348c47012b2363694c6f5c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "ae05b4aba340d5e25e205af42853b06d644acc6ea7eaa723af9ccc9dc47abed8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "31a8b855bcee943a60225578c57b4ab83dc569ca56bb419db08a26ea6ca1345f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "abdad5fffbe8260f532455f0e1fe914ee7ae0c96e1063f9a0a97b98b387d14e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2158c00aeb131e1fa01a02c79749bff552f68d6d7a12a23dce9b1817ce815cc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a00a2ba710fee9547bc11d2ebdc923a13de4e8a58b3dcad17e0ea467149aae6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8e4c96d7765b579fa47f483b009941b1f2713590b7d236b85a231589d33ca2ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "191294157f7abc3aca6bd149d7c892cc2a14fc2c21dd01c7c68da852a491213f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9437fc63ad02424689d7c97845dd270f9da1431bd0fb71bdb2ad04589accc1b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "e87824c4dd2759c9324b273bbdf2b1392057b64ecfbc7212134ec377cd76e119", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "67d724b6c2761b604fbb313e98a13dde3c9dc6ccc62506d77ecb8e38dae4269a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "99690140751f11c9cfd6923246d930442de1d4414e8fe65db4b71064fece1f50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "46b3da7dd9b32b7dcdb8f2b44d250109d3aa6e854e6dbbfb1e163dcd1bb818b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2c8bb472820327311fdf423da3e13ac64bbc72a1d219cefac8f5082f00febc25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a2872bae37417ae764704a1751484f434bcdabf922c5e84fbfb3a4d8cb0ab733", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a8c4044f86afa19e0f1a6e872cca708d68c2341c1eee9da0bdf72623fd7706dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2e6bb6c473f41b56687f2066d07a4943150808164063512f28a9dc57b3269554", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "666519bc7ea01f32a7d7b22e6c88e5caec6bd191ea694bc97c4a7998426c226c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "75e3ad7de3bcad9af0e3bc1a9227e5dd5ac9d2d8d0c1e23ea149c4d9c6650af4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "130d4ac2fee8b88e413684bfb142f623c474c95cb2811c5f36835147c6cb5e6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8edc9f097ebc231e6b082c7a12cccba819cec1b882e6772bf412caf9950bd358", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "c02235161c5f86794d3882acc95bfa6d2e2def44ccf5a45b825f479acf2e13e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3b92883efd9c82cd7c1c14cf605fe03e382821c436c930446698ae46156b96be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "a5e111633cfa91ff8364304a9ef56dc3292a414b1313991ca7fcfceeaafeda84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "5376713af44825edda9c05f1f2e1e455ade26a0ad4e973ade055f7eccbcd798a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "2f038d6704259080a3c16c1ae80600067d7c866e38d7b5b3b9038e0356670298", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "e81e9d37e27d66dafb9c721ca9c860a95a9ddc90f975ee525cda2faa6ae51f51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "edadbe1855106dcc7b5a9a1aace91dc07cfd7eda9751530560a85f835d1245de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "35bd2c4c762f6e085a674c63263f8650938fd79806e8ba9dab90dc1187be592d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "463533668bb5cdd318dfeebf62ac24eb03f216d772b7c98c77fc1f35d40a968a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "65d37b2b2b7579577ca384cbfef3b4d454b6724e1f49b4177e669efd464fdc4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "669a3aaaf800db091da0e45ddfd3442af4d4b50c1cf8eaa1dbdc3397fc342351", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "0b403f8de29a6fe2edf37fc2df6abaff1d727d5a17c5923a0ae686227344026c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "06f9751221c16904bbba89920e26be54524ee7d130040e9dbdc92a97f8810361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "7bf72e26b9813eefd31553d28d8df4c586613443915b926ad8c67e2eedadf880", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "af012636d72f2ec2b961bce9f41bc978426fc7e05da8ea5b0973744c7a59225f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a1bcc17b938b93290cb7272836b149921f89dce9e40992859ff1cd7628fe6401", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "62c6dbb8e66aa8738e392bce2b7b88d5f65438e663c2608b222418adcf616988", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "134decb8e9c8a7ba5e976b9328d81c6beb000fd6618f9135896748046b6601f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "40698b7591ec40cc9f173747e76b07c3be6aedbec6a950e302c64f4a0ac20e7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "0c94405c058819575d2915ecf78643f74aea88bbcf26bf5d70d905fdf277baac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "1aff791cf67e2cb2cf0362458fb12e68c5c0f21c878d4d03a13604d58658a4ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "8d3d3d71c1efeb708bc35d56c7fbdf0b63631b5cc54476a348e04564c7d9527e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "370f0d2b4b72097727ebff6f4a9328c5154855b157d05e997dbfd447a6daae76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a2e4b43dff94b583cdfd369b666e524b4db0ee5dcee88f5e4aceee6de8831835", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "9bb60b1063271e5243b248c49a0690d6a7c9b7cca722de8333470d5620811581", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "143773ed2e9b716097dad161def8882ebd21f330ec012acf32d7299c59a273eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ca596cf62bee42679bd2fd86323c25842b4329bb5d215909e3b4d16f5ec5511e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "ec334af4671d0bb60b4aa25427ee996d03badb05f9c19635bd18796ca572afbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "af67965b8ff7f73a8657aee1d9e32eaecf8b2409282caace8a0b3116a9038644", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b76917346c9917438e11bda9e7df2999caf1cac1f634888764a64655a2c32975", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d000fe3fbb01ae5e948da33ece2a10c4cce5683da06de4ecdd85cfbc50bf7d33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "01f187e0178f22992315edaf1d9323049470542dd135b12b72ea4c79be3899a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f873e432c34b68dae8529118a783998765e3daff2351022eec0269ed881ac5e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b71d6675ff11e91f5467933f8f2e5cf1867887d2dd8e4f02ab77565203c7963c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d375cec520fb8d8f1a1b9c1200976ce990a055efb475896d2fec39f594a12a08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "31d41bd753af5d6f27da1c1662ce2b7ef716ba95afa66661c527ebd09c25f2bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "021679c568ae63483a1833d6332b39432e073a9e86506b01b8348831dedeba41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "375cb7b388fb5cba4facbcceee6944f6f7b033eef783960a04ff2f475f5f524b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8ac1bda5359215152f74f1e69836fe2ce55b85e02fda67ad0ad95338551db354", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "9e44d4690971659af1197e9244fac13a743d6271312a7c77c67afb0a1f977911", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d2554e869c470c25b1539f3c330cbb6bb5067e58f7e0f97955daff63b694708e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3c509bc77d78c94be0fdd9dc44fee68f6440029f4b2d7dede62a1bd490ed5529", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a4e647a047cfd7e235f63ee827d8646e36a5133cf231fd40c9baaf074ea3e997", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "01aedc9f91ef3df9c5e73aeab013ab2928e4136a3fd5c43831bb94a4a5a310fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8d122227bb02507241522cf4e47833340aea7dfa7aa9c80a917962ca6d8b7cf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2045c636aab95d22bfd9eedea61fa73976c529cf1868d2834a2d6763bdc37b18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "fd750630527bf157f6b3d02f21b751874737b9ed1db57e5e1d2d3ecc2a06de63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "88f4339b7e53259a28c8373e9d639ad679c82964a2eff980377e4368c966fdde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "4a8fcdf11348a7fc65991a7496fb40179c817cfc6355966e613dd4440d0be984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0e4438e9ebde26b2931ad15e3ae57a40cd217309018839844190b9ac9e943d8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f7ea11391f45f939968cc95decbfa03bf29c521234e84d2e4a07f6f0f9f473ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "37b5c89f7f8470c1f9c0142d372c43f6b56b56663e2bfcdf31ce492359a64367", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b5e808c9a489521aacf7fa69e6780668708350a2cf93b96bcc0c84fd5434fc71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "6f698b61a40f80351360fc034e35dbb3d7215ff3fc53984e9c2bf6de107379a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d6ab1e81c0bbcf3517f2db409d60d06d152ba67e47b236b3cbb2c7d2d071ad2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d55cca8915cd2a948414e868e65043ccb732b7581aff343790c140459d6e48e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8f0c218334ddb43871ff49ef6c84b5ad5d9ae6e1a4b779963ba838628c55ff5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "df6e0ef75bfde7501dfc1e722af86b42a18513195219c7a7750ee50a1dc4164e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "681bbb4d861fb3690ce727ac2b9aeaac2d6518a1b523d556be7a4f66486480c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "650c2124c323d306018f21a95fa76227004545a74f803618a9018d508b5db544", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "dcd9a9b627d0f4afbdd756a64c4f8f8a3d422baea919dc91b3d517273f597e17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d22f6a8c6f807b96cf1a3d4c0f642ebb94b06c84388962dcb713547978a3738c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "d27fe8ea5c31f97ce9aee66a1152f6b9af1050cff90f265eee7a76d689541cab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "e857a09b2485533adad0cc367429800459acda61d7596c9cc0704739b2b47c0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "64c199d9b6ad7bc12336ce15592e4087a5627709c08d95bf77ce8cad335d4278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "20c9dfd294a41d2b7042eb4c175cc19574e594c94132c35ef28e9f5751beebd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "a3633f494e7774269deb87fd055725476c35d3140c9b6ae834c25ebd04a7c1b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "d2156dbff8df8c18eed49a2a79e9dd0711e3816bc2a70d5ca574c2ac9e3fb444", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "e56ff58d30eb3d6a10838dd498a08f9e5b587816068a67712016e26d15e2a3c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "27721d732d2e06f2feaa0137eee1c47e2d2e644c269cec89ba3c5fbd0e142b07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "1f9a72d5b0cf1f6cba0be43fecea9081ee1482bb636eccaf772fbb7bcc3fa846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "f352aa4ad994f93c3a85b1d81e67c5ccb2f2aa797b33a3883ec46ea7b82f612e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f4c8eac9e0ddbb6fd8b4723fb58ba8ff54a3c459fddb8b67f659bebcaf1cb507", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8c9fbf8100ab31cbf3baa3da4e32727f1550869d7cc15392a2ea4970478c2e26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "c9da229f8711af195401bcc9d1060cbb7495e2a4931a5f3e3ee4bde22445eac9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a5c1c8eea351088cc72221b5c715712c7693ef9d90ba9c3b1a69765bd139cd94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "08390b69b9b5cfe3f217b5176ea98554ca8c2f7f3ec86ff1a9202bb42e8b4f07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "287e5ac22c29454770d0a7472019a5ef4f7e3c1d6180456a389f25c418e25512", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8d082102ee2b37be4f538914d7ca1ef88adaab1d24242d77c581ff8a20435534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "834d933e200fc43017cd8740ddcd285ca1a42d3afdf3a70292d63b3c2adcfd93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "1facee5ba44e794742ed7ddd1d3067dbc5b339d0d41b1a502c7031096fc3a5e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0c09bb94d15ce79b7dc9889f45c4834c4128df463f81413579fd4212708ca7aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "50017e95e1b2e25a3df4619c6187ea8b0b0cacd9893b38886fb68056be20ba7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "b5e093e7be967d04d2669842c6afa364048be3b843bac6905e6568498e1c4ade", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f1b566f23a709d5642ac067f8b537039db3f8619e32621691af9f14d0f8dfdb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "058a60a815117bd46c0ab51696c06aafd7b81f5c5d26f79be8d183dc3ecac20a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "264f7908fbc9933976059731084be2a967f58bf6010c209af356a1bd904ae394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "b71c771a27aee13e5f8fdb8871ca907b6dd361b364eaf4f8cb404270ffe9e59e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "83ea9a34bd951ae0f8be230fc538a2f0f61ad41507c584daeb693fd0dc53f771", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "82bf83ab017730c46971932795d8b1b135901c36208286a12b1c653b93df197a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "7ac0fcf28b1663fa2d517ce80342546446ded7dd82c06c4f2ac6f9098960e83e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "a94c47c45ab0b434826ef73184b4f9229e96a0286893a7609918ebf0ebce6011", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c8d38a2b7e6c39a6580939f62ab8d7b790e13f1d001b016ebcdfd721daf649a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3ca08301f321cf8ea4c4470735574f3576317ec06a1c34f56150d9c297403456", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "1f40a20b6268a6507bf50768d0212d9fc4b1df41ff7ac49c3815fe420fd3ef8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "daaebaeaf3169bc3e9e8e6fb4e310fa6d89700830166bcd0de6a0a6c4b12ef20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "521c552ccbdd42dbc50328b505af4ef143921080b2d342ffb5825853566c21ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "6a8f81fe8b6a9531522d28a4fac0250d1f9e967efd534264df4bae918ea44172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "bc3d0d89669bfba791ab6e626405b3fa7a89d0ae94d324cae1e39af8b5c73c6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "de7d9687ea4150c1a46bfc1fa55b34fa9c3d19d672fc56ae4d2d97bc5b1dce1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "1c97ce58213b2d708263cca1266c2a36bc1e246ef52f0ffb9c78fea7d37b59a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "b148cd0312ccbdb895d3a2f28d019b34efda7216e16f6a3466df6473cd93a9e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c3190f303d8ce2c2225e094b4d945dce47ffee397400cd1ea5e6b83006eba0b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "b5bfc571580ab75ce8a515b1919ab8209ea0c5f097fe2c5c5d44f9ae2262e57d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "b2d74f74ed96b4cf73df19290274ab81a3105c2f0f480d2ea9b94450818a13a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "a4bb36f2164f5c28cd2f0e189492b7175cbed6043bf5d31a3f4d35b195bbc0f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2e4ab2b6f63d587d9da96fed92b4adc22d219e1fb5f41f2e7d90e592184ab383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f5bf9a7ace136da9036d9d046bd51735bc54aaa89a2acf818e1d1e31e8d9ae4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "94755750b66310e6c61ee41c0e106aaa801b5bf632f7a7eae41be8117ccf4a4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "54e6f129d5af3251970d8af176418f9e90f2d58d8dca907f86252aeb1cd49d9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "30c9a5bd0f0ec3c456d197104cfe86477b58dfb7b3eb5495a6ea31a1888ed6fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "9af27e0617b060eb9267365bf34c2f39f1b3c054a9a296d2d2700c637d1abd96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "291e0b5b78b7762bf9e298d49d0b55dd9c1bb47e4cf94b72a1c7162258d5b741", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "dc88f7cdb5612735975ef3dd44bd17d008dc365d955163f95bc2fd2b94a10d34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3023fb7282c2b72bf0da20939c7f79c8a8d87429b942aa34659c93dba21e415f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0427d618ef8bbda0698d3714f0252ee3478a581130693aadfd124803e9b8e362", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "816df18c119b774764fac39c9f18c30b5bfd00919c013e32d85a3cba7c796927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "6985284ce2c2fb56666dad706415fc1b2bf6a5276dc99df4149c0d7e74a79024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "8575255ac4331d4bff18aabd4fb0e8fdeabfa7113441823c0d56a972c855cf41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "c5c72cb95a2d4830a52906525e0aa289c06d74cd2fb0461dd26e9ce63024584d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "1f9c3aa486e58ef1c6926afc65144afbfb98244ac5de93e6620e4654ac67bee5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "7fbd1c7b4f48b751fa00b8789854459c8b25509e03033bafe563de9ab49b0f9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "d9ba3f9800f439e7015848d12359c676a67f2b960e976f410b80aeaf1fdf0706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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 */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "b55d80fadabd379417e61ebd84c315d9474058118aa25e60cf37d653e8a5434b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "dc0c791c9206854ac1b78c6e9fcdcd30568ea135a6a2f7ef591d20de48ea51f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 198296, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "5bfd26b847293bafb1f7b050799207172e3df817393d7d8f30b04ff7cbc9f1b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "1161434597d02fb0a114f3ee23d6f6751d435084ded996fa09a403c05dfe6425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "2315dd4d7e891518e21198973f0c2116fe5f844c2dddbf7479c15d0be6e16dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b03b9745e447b087c6ff70483a2280f04c9c64f96caae7a6077b36d0d4dd86ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "05f73be9bdf7aaad63782f183698a7f34f2bf809ffa9831cadc3faf76dc6ad89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183952, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "32ff2423a8f0a92698968567dd24081290266968cf130ab1799773275cb7c3f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "71c800a673cf5c1333fdb4ba6ed1bf25f9bdcb6c0ee04fadae8fe4de33c2367c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9783a3f54eebff27f83f64a77490494e4922375bfd2b8d04ed1e33844db4c396", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "35955e0f352ead8612c2f705b82875d1b2848c5906bee241461ea5d577e8b104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b5ac6fa8676d57e8c95d535f7520fd66f34b4688788d3f8b014071aedfb6964f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ac1b96ff4da375f3cabe3895144aeac34cf564ccbd855301710d9155bfc488a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ddf5770bf8096fe98c67a602172bd1d9f5246deca30eea52930ae2be2f92a3c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "67a9a415ee0110a578e0ce1e7515a23a24eeaaaeca48d5065bf830dcd3893450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "bd6a716574715cccf7ca89bbdd3fabee9ac2531cb41904e5689ef9d7123762df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227728, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "1e068530b336503161f0c44b0f6a1f5cb4a707c89f743017f567d4616fc0c189", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cc6bd823603d0a22c037b71231966952261e381b31b2a65fb4e782f9d377d55e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "d22ced0294d3a96d6baf46f97a699ae7f111c9fe77dd860b39e9896eec0bb437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6bb8c4e6ea10451b7fd651aa8a872c53b46ac8f228233400ff3536972902b4db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "16f06b7b144e16985a8176892f73a4cb59f6c91db2238dd1cc5dc5948d5b766a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ 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(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a1bbad4357c9f9612c1000eec80b59a74506ceb2902bf5a18fd34a6633239dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "786f96839e5e7c33c396f3b464e278bbf1ec8c3f98fe0fe1ccd40233191ba68d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3825a8a099963ae56e294996721e2e4cf468ae5e753e7512661e6f434e7fcd8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b44110ad68456274fc6f9258793fd7ccb5e705308e1e65e090accec40bc4018f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e621bf21d99964fd919b5d7e77c07010735647da7ec10d74effe5246454f5469", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8eae3723d631784079c511246e0a70c4bbe932534c9bd1e49fe4e84c0d987927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6740db243340c7d17966feb13410cf22bff187faa6c3f8386d2f78594c05e536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1a0ea4aa3e89ddff9812e024b84f86a51fac3354467cfad0cda548e3748f7854", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "83448f2b3e24176ce96ceb0e1c4fe91e8d98b95fc4fc35d4b1966287f864e20d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2da1575f2b8fcc9911e514028273462798f9f5e23580027f2b6b0c7ca0a5def4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c7d34429e61bc042d5381638e80234d9e761ce67d50db2b1aa3191b9a9ab4ee5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4c42e144f8601fa91da5748ac8fa5699faf4e77d746d8c877b11b951c1d57657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3c96bcd17ec311275a194ea400d0e24caa782aee58c311e99fbc5984aeab33e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1de2f2e4e0d63c89fe2816a11b43ebf95fb45956a29f83906c3b947cba09fb7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "214db03c623cfca958095e7797ae80788f364285cb18c1fb15b87d8fe655ca33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "aa01889fe7871aef1300b00edd049cf23264eb43bad46d4e2a7b527070f81482", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e1fe16191283e76477c9a6c69b7633b34f619a53c829c15d838b9103f539a002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "88663a94aeb91eafa03f398c8c214cd50659a3e361c58acb80a0794e52aa9d8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "51f7b823be7e1549e81a970b04cb0aa0fc3285d06f733f9657038414062586b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "16b226a9af94fffaf8442af2f30a8dedb446447cbe10812b545e294211dba087", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "92d628444014b173f924b3d5991f44fffaef3140a8f584bfeafc52a324ecbd67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2caf603dbbbf9c7002fdb1654e6654413f6df99551f73387f852caf6497db001", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bc106b74f861cd7830745f2917732decef332c4f88046e58a4eeb7a11f76e3e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7c314a51f17ce39af565703572dba93d0a0b9ab30305f4ee14441305d9655c27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f19d3c29c085bab2fe8d2e20dfe95cf35e770f8da06d895381fe07eb8674ef8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "12137b9900bdb444b17b5bed519eb938597b778ff6fdcf2849776c1508c57eaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8ad69edbe2d01125c834d1ce0c44db082223ceb8f8e62dc7ecb47590bf81279e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "be2a7b0d4da31ed10bd5d0d268d3a56eb595e0445e21d2105bb1744aa57a4498", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fbebe9d45361a9effbbf7344ec4f81ae074f3ad994868328dab655362b9676a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b713b6f1e2a980717c33cb06e02cb79e8e0be54ccf3a2a079f34688c96b7deff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6faa872470499d08d8153481500f45b63d2034373274606659dc45ae300a9fdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d44756f521a73cdbf3fc66b89697d465ba479a5b2693e84bb29a1f49527123c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3e72dcf447de2257cefe08efab90e73a75ab787e1d5c1b010b26ae038f25cc8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a4f33158d662fb9900ea4fe7ad16841d80122eff8754dd21d5755092bf6ca4b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2fa5f242e9e21cc67255cb4f5db13444b00eb70c43ac8830cdf47fe73b200b62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f3e034166faf8a0176ecfd399624194016a001a20b9c458b6857c4ee7ab80233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dab53a859027675ea5188a7ed2bbc3346f15ad86f436ba539f73e6060edc9e7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0820b5b54b67cdfdf8c69ad705df977c66b08b837e7a535c83622b7a872f6e53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "99d28b3939493883b5844d6540f235dd773456073502d0a0979cc49257bb3805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "38f2f4acbf819d2101bf0f0b1a813dcc004495f8390dbc99f8df3c89b7ad0c48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "59a4a07ea8f4902fdc19c705def988fdc4ba21c9037a98ca8755cce35bf5117b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "989da12d4eae110701994145af337825b0a11691231dfda3cdc8b8f3be85ef16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ec18d0b35c34fc78da75a32c86bcacbb21ebbcab18272f021176a3138f5d96cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "72f95947a41d504f8353885a9aa1a19770a12b22118447551b99f386cdffb227", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e194c779b6ff8c923360f88d121a61d9258c58407c447ce7b9fbbd7298fd68b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fc9e8722298a35c87c8434e62137a9d194c88f90d4eeb6275966fd1c7b788115", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3a64827080cc0a665b90d4f44060bee9a1a0abe761a52b2029191ea8eb405d60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b000492fdd7f579bdbad5fcc10bb90cd2dcd7516e5e42868d82286efa2ca6d16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "50382ff7c30f1ed320ecec37fab2f1fe8ac051d74420d9d7c2f6a06bb6e7c8a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f426543c6ed5ac32f3692e16305bf772642bd95742f6d5a581e3cda4ff300c8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f7f5154088e7b1d582a36211ea022c362c2b2788db3ed7a1969775ad98c83297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0bd2d74fc5d81b6733a5c01cc2ab7571b39017ba5f804fc578e78c0ce90b1a1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "95e7a01531f9268498ee9c4193db9916c68e792a712d7f6e60a1222b824460e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a469bfd773a27fcf468a591049e2e502e3e17520ef649e78f62d6b9173727b59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "179181ba7358f5d760bddef1c4935c0406e59cf8e380ecd8dac48d55a6f6658d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9c8623e3f7759688f9f99575ac7e59042c575934c752cd00198ad219c4f164f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "758fab2528b57f840a261406435e30e50fef99cc64ead19b0f951702d9cb3656", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a658151d69e279b484f24cd2211db8f4b6d46b8110440bc176031f654ffd5abf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a2e10da433fa3eead442436afee1616ca149f9dc8efd739074a04d9ab876196a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e0205044781e8ceb934f5c14482d9739f09c8c4a78e118a94cd3d4e3b207aa50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "387a955125fc0c2237b100f17d1089d775be63f363de3274a91b184e9a6210b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d3041e2610edacc5267202a2c2220c91b6b9c75c4205baa0fec8782e790c3242", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8e6fdfbc6ced0fdd0cb15fb3d2e58eac0b6c071e24ef7bcfda12c497b9258a43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c829873bada6521dd056095254fd0c302e97fb61fbe090949c10d8759a75a437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ba6a7a9d5b6df840645657b4e27e3954582ef10dbd33019b91bbff0abf6869a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8b29b7e1c5aedcf671c1bb01547544107fd945d2d6c9e81e7c15a3c8f6c788fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9cbf860aa7a56a99551549e2239254894eeae990f801cca32fc52c87c73faa51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4e6bb54c91fe0f5e8c7e38b4eda47eb7aed7a5ff4f27dc9ae118e46768445b40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6f5986c5ffb697eeb62d59912d6c5356f4de2ec1a219bc008bd0942ad113c9b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "900372b242aea6fe2793a568a44bd47dde80671c0dce23cac7056e6bd7a49c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a7995e5215fc1e39f95234d546695c17950a3a0010126b78f3d6656400ee0c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9cb02ba7341c6f42144a91310bc36ec90d3dd1b46ff8993b38813f26a0da60d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5a46b2095c57bb73f5456bf3b38779ccb681a8cb6cf07ba8dea00ad84322f6aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "225625bd3cc5509fbce1ccb7ed62bb60ec77770a1e4f80c3448fbb500d19e0a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "12e62c5874d366ce6b5f55fe9c39a7a8dbdcf56878abb842793b17f2f8858af4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b48353492aadf77e0c09f12b35b21543af6f791e6cbca0b7ab5cd4a478ea6c72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "05f4aba326f91ebbd04a40066aac736aa28020d684aab191846b0d25cf88d53d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f74f1d9bf2ed0357e66501ad3535ab8cc4ed45917831307b0f135734e4e18201", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e83f0b77e0e8812966f8c5c6bb07b2ee5401f3e4f0c4053279023f8c7362eca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dc52b033681f5ccaa13f0b699751cfc6d62ee2196c63f016282be9e51e1ca309", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fa02184e5e5591299f23bb46fc8bcd6ec2479c5880a3428fce42fd8104035861", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "95ea8886efc5a4cfdc50cce4ff24409ba0a39d2385d39efae5125a1a54c3c6fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6d8ba7f7870b7879950ebc067738b2656af09a52dac7ef5dd87d4339aca57e8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "65b435ca1bed6c02007de7a71747d14c6820c4c9bb9e4c2db3f6d610cccedf7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0091ea8490d802b28c07f13780daedf5200bb71f57f6ac37fd50bbca6a0f5271", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "23be33652569952cf097b6497a7cbcd190db49b422718a33e9800e2d3d9735ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "829081ef0cfe79e9768e816e6eca8970060da214eb6508284cdf635b6c9a98a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9e609e0dded365566abec6176a28b2405052e6a30d03c7954431d65288571394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a05de2cdbf4331d39c28f4cbb757683b5860c004860a5c14395968eab1c5ccf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d4cc94fda7fd92d5320ddd82df6b6adbf737ba6fdddc1f0a4ef446cdd23d8e76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "883106410dd95df2f14813d3f07421a6e2bdb49eba656001877e2b7f5de3f4d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d80494ca81aab5462fade34a60adb96e2264dfa3fa689b39ad84e724c7686bb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1324d6b8f90042b4057dd7b41ce537b71c9c70326ec9bd8a8e40f911b10805f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8e97a9bb200bd5636a3c3fe81f2d5e2334a1221bcaeb8e63b3f36c62f69d14d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "558e19418aa328e09f3450ffc8d25abef56df4c2bb7d8456bf5f56596c8200da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f2a5a580afa64ee138be05544d2cba306152e7e836f3b2a2f97360a0562e7f17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "068cceb2c373e1f971def0e781a6b41cacb5737f98c7557cdbdbffa4341fdf1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a3ac66ad7289fc2f68d8c6ccccbc475156579469ab942bb23138837f85c09cd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cdf89ccc65eb3bae26fcd454aea4b699da9bb179533de7e986ac95d1388bf5fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e9185900115eb5e474c6175cc1ed4c90ac1cb8576ce48b47d348fd96767eb207", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "204413e5476e3f82af6100456049ef9f1fe50bb8373b67bab0b7ff9f0398b175", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "4b9a8e2b5d1591285bb3dbfa596b1d994fd939900c5cdc9ef422fc339f2da299", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "bb87d3f3bf8cea9dd8166b5c6fa6336a8ba5726584aa72ae51b0cbe055080c10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "3beec680531d6cb229bf5ef833fa9f98372b649c7af2a7fad98edb09acd271b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "7d9e5f98276008c95e4288489f081c0095d0a1198c23af9c7eceb9a3108c9cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "3f9215e354b29c602b5acbc568e8946573358b82d10defe92fbe79fa7a925399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "8975480ac32494e255b8ed8bd491f9a5ce0d97f9f6d5aaa214a1d6bb45d2296b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "e91787043e9b8aa48c6dd90ddef53d9b76b44b997a6e942d9857659c72708eb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "2a51e083a871c498ca7046071234ac81a5bc3ddbe5e28a58f073a9776e2d0092", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "bcde729154401ce08a248379d07211ceffeebf58eafebd1e1fea5bfb9e7d900b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "ed397713d9d17243a6a849377b5e6a60eb0ffb69e3f1a53c9a499251c744aeea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c226aa378a3001786ea20d6d3435d1e05a595dfca991feb56cd145a0fb29f05e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f73f8520b605d0a1619f3535ec6e519fa6a2b89ca7da2ef09d806f48fe2cdcc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "71756900bdcca72f92b90ed365ee2d6280bf012a376d4c0663363d0fd4dc0dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a71426034dad3b56124317b148cf6cf7bbd22f2e588b1b15d7119e89f9d1f503", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "480bdbc73a82fd9c8032e2176a638ad5772d3196f90096dee3d7b62e5cfcefc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b813d142c5c5197a8b6370b7ec2af36e5d6f19ca0d84ea89353590d5553da3dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "755f60adef201183ff40023d5cfc0fd1426d02ee8995869da7e96b6eb560ab51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "ebe507fd5fdb79fded20dc8b036b121afe22a4252f22e31a08e6097ec9e30dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "0f3b2a8cd5b3a6d2ffcc0463db289b1fc2f138b55db4793746de56b3d0347398", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "233f675258dc4aa57ca152752c2c9b0fbf51ab11e4f0cd3ff6c9b25fd78825b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "09817adb7d236fbb1f1933cf42974c2e0a8be099c22fa32dd9f3128c70385de3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c797660da00d0d5372b9c0f3182447bd3f7f9836105ba58e79795f09cd5152f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e6a6f21f473eb7659ab82ceab05d652eaf71d2ffc6592572f5ede9a716abb0d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8d1e1c5760bf7c01d7fc1331a2d2bcf6a6f89a75d6312d1affbbf19234b7321a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "fa3818c9dc12585b87e4b73ac04e645bc617ab2920a333788816c95d260a1d58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "3bad5e4e86c75cde2a6705740b0ebc9ebc60b6c2981571e48132628e31b360a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9d0595be480e773fbac557a617fbf1810dcf42b301a0ddeeebcdec6ff7e4f1ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "5acb2546a794c1ea578aafbc5bb262bc4d6acc18e2715a685ef3f67cffd3b2b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "be975cf675e109e67a32c787e6af879b53941cac16933e519fd564f4610bb385", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "ce147e78fa38afb6df3e7aa089e464ff1912cd005bec9602787ba7e57cd8f494", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "5c33055540b80f183dca3a06a80ecc1e3ac6bbdd1051b1928c46e1a230d0eacd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a0480b3c82d5ccd3dabdb159acae56e59defaeda26a73621081bd4b352fe75dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "05dfe6ac090351e906f518cfc521717189caebca75f934503a15024b438d145d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "4dfe66c7268d4e31d4317cb94fc9e62d856ab4409862bfb9d5b41a7eed755e3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "8983428321835d3c2ff674701f9f0db6de7e5ac1d92f78d8d5e520880c0a59e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "8c991c06e9e65c9e2b0702153670b1fd908fb47d6129b9df7d41d906a94df8d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9ec12da2b448c953c55b9478932bedb2a574f7f3a10252bcc7f6dea20a71119e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6705d4641cecd60925b63f8ee92caa0934e8da850ac12cd16a50e5841aa57058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "cef4d4515cf652bf7a4af1b99f4a8121b9514476c9b988a2de714d0c0dfade0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "d6da264262d6edabe95023edea3b5448b7d80a6e267015b7b2434ff43ea81ce1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "d240baa7a4c1059ca0908c9e4f49fe6a146c3f51fe678722219428e618aa3f86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a8fa21b6b60c86a2f303602f79efbfb132909da3297245602475c164572debd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "5626636d2a19ae0ee753f51a06a464074a56645e08840ad5987220339b5139c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "2521ede95a79868c7fe379717ac41b7f85a0f91abcb5775660fdc218268519d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "8949d3bfa5c46611d1c4c1b466f29519ef5b9c5efa6894f16d61930738388267", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "06bec4e4bc2adc685141e5d3215edf0df356f8ab0df8f7f47060dde29f5cd7b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "208ae957fc5894ca64b4fd3463857787105dfc4c3dff3d33946e5c962fcef003", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "7006d026059cb868abfec8336d7baf4eb811fbcdb38e9b52810b61e7393a02b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2370a4ec78428c05d801d8823ac640c172bcbe0f480d28b7756403e39f03d34e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "df222a4c7c29021a5ddb37b79cf5647c157c7433630ece0791034175ba6ee06f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "95db755977fcd7539cd3cb87d49546bc484046b9a71d1d6cb789c21570788dd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2226ff95d0fea907bde10d8709029a59f91cfd3054dde1d520055a06dfcf401b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "aee46b68fb2982f8bed687a6aaad6bfd3b7239e0ce9cf7368ae8dbe6d1faa199", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "88673dfcc98fc670499ecf0b942ea0fe802ecc921e7c207adf14276d907b2660", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "49d0aa5076be259f2dcda32e77be3308df8dce182287be13e12d58fb274f6daf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a7fb6e13548402a1678b0f73d94f0c88fa2778b36ee7966f4f20f103887ac6d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1bb468502577b166efc67578361a1f48b2241cafd974c063cbd7bc9376eda790", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "0d165a6612965e3f55b6ba82d1a16a58b347073207680d7181b625d7a78c3ba0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "13608ff174099c000db9b2c0a2d4761050e68953ec2f581a3d855d533f5a9aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "4e9a3016a7fc02227e03d2a80e2d228709cb760663084d144b16f14b9248bb06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2cc44684a2b7662016ba12f2c32e2f0db1498e1c0424f99d3fef3e239fd7b349", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "04d0b013c5541ba36c4528aa85628b56de444f4e4e754f129a2ecfccb4f078ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2b0d02f92b48cf168bd5c73975ec36990ebb182f2bc21164b07589b6dde3b006", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "35dc5dca9d3b0aae21407e3f2788cd17294684e1fb9e26de3a626079fa2a2a15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "806c7ad05e161ded04aadcc268de016a17fd565c8e1457f532e9be2238ccb173", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "4a988353cd818229bc34dd42d04dff11ac515360536f1ed1c763fa0614ab9f35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "239ff46e7cfc9c7c82c20d73dcf47810a14b238541b39c7892821322bb8f8f8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "96ecdc33e356ce12321eeb09f7e8a61be063cb49be3e1c42af9b183bce054c56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b69c86294f133e4df8c908b2242c4900207e33e35d35882ef714d7c9b8e39518", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b13e2ad865d50f46d2105d206f62eeaa9de0ecb030337da5a2f1161a0110b0e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "efffb704705cc0aac35b306cec6a09be06f886ba602a0377be4e9f54d10158b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9e4e174bde1adb8f1b36539625ab0223e767629e56a86e8d26b5331f7f37b63f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "631c260876e20fe3d2cfbaaae058d1c32a2882bff6eb55211161953ae19a21ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "5574f7e94504077b5777496c559f358bca15e3a8772127645bbe8df88c512e06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "d0350ecf67484899cc0e9e5da984c19129642d64b75334915610a30e0cd81002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "e81073dffcf46a502f29e1edcc3bbd8adb014b78b816244b128ebb1d909e0bdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "f484df4e521b6c2eeea6291824e6faa4bedffa982e6f66e4ec60de1b1de7149a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a5c8b578290230575c3d421334273e0872f515b5112f538745695d23c8959236", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9b7c99ff51358cb1f8cd6db59bcdbff9ecba7f676903d24577f6b2fd6d56a269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "70bafa96f086a9a4d365bae9bd400ab388f21a9537f9f303e71083ea4b738004", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "cb9bae339e556e3a9bead46a67b4bdfe0f546ab166c2ef230ca655656f77c046", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "278260ad2e51a651fbde7d5477f6e0a49d94ca3118351d1d8d9ec994dbe07fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d0df017106d2218d3d7669fd74170be71621d03ba1616efdd516c8eae3b99968", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "ad9fb1220a1a0fe2f1442552d426ce5587d0e843810f6cbf5ee92b9221338985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6400aecdd30192d9ac02095658e3c604fb3b226a545395887bdb121bdf003ee5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "99ffdbaeb3d347ee12886c46955c17fb311ff8859aa0455b301d7d51e1a70898", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e19921e1df53c67c3abfefc99af1b1dd85c501e05635f0605b3e17a63f1c74c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c2234ac4c712b3a87b0bf5d3d769dc4260eb5afbc0738f1f7481cc0c2e2231bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "e09c96b84540ec6417cb0723a4619cc61059118b589e52ece14c973452bbbbb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a77ba803cbb981e0f9073bdc083ffcb7159edf8e4bb385dc62208eb95a3ea0ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f15608f8187b94f6a740c18a021a9321c5f028cd056e9aa5cd521e73980dc420", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "fdf0c7031160f35720257caa7e1be9d193d87dd2cc1e92714dd7b2d8c3ba3e4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "6f5f0a0f620fc5ca6886d1e78ef529b92a07572ba7c14762f898b0e02e8b949c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "40b680781319d67ab69b7cd82c94ef8a9d5d3788efb0f88d49741498154dadbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "72cf3857a2cbf02fa3b32d267320ea5c17e71c55480abca834a0386842d8671f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "3fc0fc80adb3587150ff3870c714b2ba16348de470b05f603e3a8281275a1384", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d277b84f1ffc5ee32351f8d09eda129f4ab8c3df273170a20d94f7351db49055", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c0539115718e5f36b52858736912be272d865920dbc91e65c42eec290fac20a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "98e7bfff35af758850b2ba86c3673f7b79ed320e5839bfc259415c6d396da062", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b677967a7e3f5606541539ca512b95bafa5e91abdb2a7d1ba4634daad7eb08c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9772699f7b4f2396153ae45f9a314761fa377a3995e64fd6028b0585bf97fb98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "711c9f806df7d1385d99e18ea5727787bfb18e83bbb1cc6d9267e4b38dc46bef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "c0d9df2587b61e99922634e1f87712d06b9d9ab29d08af9084f501f793ce6ccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "165334a7f091df9842f2f72e13711e6cc3fb8b6d72b9ee09fcd0e467851d620a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6f8ebe2be0822f167e6aed69e4fcc7bdf510254dfe6320cf7ac843025d2ded68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "86c82dbe624a12e3653eb4ca6f03c559bafa8eb64cb9a48cffc636f69e0530c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "7f782e660ef73def5cecaeb8b402c92c87f3689454eebbe503088e6ea9339118", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c58c8ee0ab4286637405426688c7673793270e35077c9a2a9d7234535e3b89d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8022b3d34ab26c624fcf803516018494ebc668d4ba2059d84e217f8cbf905b41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "36db8ef1e50bd3a51d8575c16d7a0d2e6922f2174277c9273d8fcabe20faa4ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f61e308d63de1fbc4bb7987021073ac4ba19ffd84b9c17b6ea2106f0c0a52237", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "e60e481cbbf89bcf36c682905d926a57f85860718ba184e80cab2560969de3bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "3ab7585e951e16862293cf294b9fd22c03d79551e3afd4c31628f2b49e8aa5d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "edb69ff945a8d45f4f4a998d4fa3c8455621ed58f58b6a0fc027ef614a87f509", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "5071fcb6ebb9d1cf6c7351e0f3cd2e912c8bb495324c101f84bbb9225130a7da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "51a3a4cf896563ff927d8636e74514ed2ae020f7af5360b02ed20a2da58a7638", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d5fc0e162e418ace141564d96d04515f6f3b9ae70b2e4aac21840d3e3cb40271", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "adbb6150213d0c611ded59ebb147691cf2e5c6a11d61eca75a6baf6d3321dc77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e664c2cef9457ebc0094edff2a3ce1821f8ed7185ce77ca54c2f7852083320f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "9322321340276482d1a36d84a9d72ab34506714bf75116a724ae5d4a33508d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "73ca1bf2ea8f7f64eb9cc2eaaa631e42fd7efd2946bdfcc26a445061b0f4d786", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "69727779a7818cf9842d54f12fd048b38e0aa88d72b427981ab73cad73cb900c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "115d969391df847aa99f2a8d8c13ef06c75fb118cc92e5182b83842963e068d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "2ef0a57ae9b07f1fc1879fe42505fe84d0bb9071488e8b2fa8128ee3dc278431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "d8c33591dd07c5355782df5b46030172b706327ec80fa9c4910074bc13086d6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "07c232bd50c635276300c9d4b8e2c7e75660f842db00c1f60bd6f923a07fb103", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "89e3cd6ad64319ea45cca8399e878dff394878eccdae1c46cb7a572ca26b1156", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "88dd4649b770d091a92ca9465a7a22318eb7eb9f806ddb280078b34ae7a70d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a98d035979788dbcb33a887e9b77197b97957621a38602a3aa2ea290e743c03e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "0756fe455ca18c01d59eb24c267e62fda84fe81b16527bf11519a85bd30463f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "4246df6aa79c02f415e0de4885d0fbe1345c44420e78d4e4fa7ea10cca550c6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9032b3ac507c0888420990dae751d6b75e1524b2aec51ec03a76aa7beedaaea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "c5427d53eb8d672f4dab6a9f8ec21a4d9ac2bb8cc29d455dae4a4b5c92a6d7a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "832c927ec5ad975d8e4bae118e1f4dbbe3ad0f6bfcc702165e999ce22b763220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "c4e12c9ed4bfe12fa4011ed723e0b8d92104087967323346d06f784e81a877bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "9a4d3c39bad908fcaf1094a4df2091502eb31ade9a3cea15f1efe630c902f20f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "45bdc156c5a8603c82543630ba3dcc41c3a4428645c68b128b4696c167ae3bd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "febc39eba226940ffb6341bf6814a1b2cc82876077663df76091e00dec886ad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "b6417a7ea84d1211075a44459d4631b5c4807434fab8c2fd3bfa9ef8fdbc6ce8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "e7a885a34f55ee3cab858bad2e86b8a9108c1a8fa9e6fc6fe9a313a630b6ca4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "151ca74180902d9a578a22f63edeabd3dac7aaa7d112340ff7aa06b6ef8ec49b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "b20ba2ca8a969029e4e89aadefa91e88b14223746fe25fa727ed8df6560ac23e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ab894367eca560b84032592d947690da490bc93222115ead5d72908b082ab814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "66c3d53fa60aca820e3e450f2e1a82aaf2cfd72530105c9996fd752998aa54b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "b6b223f6e30232eacb40b143899d676542671c978d8751dfb07d270c244d98e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "1045376354d75728a912c465c3b4b43f75a11825149d9717d8aeb80fe046e6ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "18fe2f395df392a45bd02fbb8d3d2ec77efbac02aaf9a09edf93319cad867772", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "a7920ac0ce2e6ac067d532c1e72777a70e52e0d216345a5112158cecabc73248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4849d2c11a8a7322bd82e0a1afcc8e25a41d91bb98142f6d4aa02a3be88e717d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "840d0dd7e439ee53f5dfdc882e1c42abc16f24284728f73a1d8c0fc3bc3e405d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "a7495feb730b71cc1c86d6650a61a5f0f26658f5cdc284b453cf227c32bda5dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "fc04381c5b8b7968d961f29a7236991537aa0eb996b5e7cdc3c200831c32e413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c6a6f9ad92bb0999702f7a569bdaf1c805db088816d407361b441a18d079c150", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4bd88cc99534b65c2859721cd05978f19d9f1736fdd5d5ebf437d49fe9f3648b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "53e10524138278011af0b33b59929ab5d8b0e0a47904ba7ea115d81432c72988", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "bdb5fbf681935547248b1dce2f35bba3c9036c113759f8f997ee52cfc8c8ab77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6423261332e94324b22d8e5ecdc7528498799ce718ac92fc467d3f26721914ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "8bc9b5a27bd290024a0dd40085e780ae764e1c6b9afaab87c7716f84591538d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "e1a507f0ccf42631e19bee2fa1e898417fea78cd772f54d3a62206ab55e43f94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "d6912192a2af379b5a66358d9c3685f72acf0ac02749a5d4fe3b01b19d872a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ee07ec312a6a04ff4234d64e92a529d172a092f07e859395ada4349918c908ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "b88baa8e1a034b50870f0e5485e1c8df6da3a74ed91c1311fb33da188f7997f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "752fb6aa7b77f8b4ef82ab6e63a646f9e28c2bc59ef03dd0d05a5e6b83beafe4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "730b1d6eec80491969f0b1076c887a3eef8fc13d0aa87695164aef5eb1cd214b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "64dbf105c0d5b7c68de98a86512e19ddb0da0ded3836c11aaf71812329c6723b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "9966a4707da11e4a0e02c688f22fec845b6365d44c8f7469d6228cbf40b639d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6867927f6265c0ace043d22b128a74210c086ae8dc938010fde913362717f1b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "5d6822ce127bfda4da73ea740f4149a2bcba94a1ad79721c8ac41ed48b6a042e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ffb6825adf5cd90d2bf638a073b717997a36bd3fa050efc6e64ddf641f40b91c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "49e126862ec1ee1265387c7385adb8ee124b046a1142458f4165a916214844cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "4534a129b4f8642201393fa4f1cff4362a371de769bc85141bb859f4e57a9e0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "b88c59c4106d0080d99af82fbaf5de23a8dcc7708c457ad667bd48d4f6b66d5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "a6dbf868c7f541e7b1c4df3f01fc2228f007f60d1b43e2faf54ff6bd03407469", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "9549110de3cb82e656a0d4a5cb2400861995f9250fc527fdc379b90c33045bf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "572973662c4fd73c39c4c36a2675f58cd4a86716c954177e723c6436938b695a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ef0aea6d5ae0321d425aa730c76b9263926ed2c2edc7f3324b4eb90adc1c45a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c6709b65404f657dcca10190d1f58aeb5148d92f9c9756f5ef7e183d0b317d8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "805fd38a7e95ecff9e236865844b7493f154c1f806e7194652d21d446079a406", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "3cb63ff31fc09b564da87c0dc4a0d9b470bb5549a0c32ea31ccb532034d1b24d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "24bcd1cb41e8f948b04f004be89bf2070716fd0bce581cdb28931aea47b6661e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6bc251487ccd07a08bf442e21f3abd1d0b6bec1d2dac7274e0d10059a9dbbb39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "5a57ceb4dd27e4f91f63512dcc983d990d53bccdfd135010044051a384640d25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "fea302af211bb7a75e8267d1a2d1582815de31f1c96c0652dd4d60c1e018272e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "79c00b5606e9f25be2afc7f409bf44dad01f28b1976ed243c03829d9cd912737", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c676199396fe279f7c21c57f82d4f3581c99312d26bf0ac0aa3f1d042a972641", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ff35eaa7c1d6f1142f5dc48fd8e48a763ae4de1407a993d36638735bfab3a9d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "5ea62f175510e5cc21d84e66de6be1004b7cc235c7cf1c572e055a0d5450cc6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "07dedb3ffb6fb10e9413ad4ea864fbcdb1b8b574668f71a5a1b56d9a350a2c35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "edf0c07a42ab5edc9d0b39c48aa7520a7fcf6acc5376b92ee7b984b1e9fc67ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4340502c99730336bd8c7ebdc42487fc82946d51fa88ed1bcd210e8ae6d8b043", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "d5262cd99baf4fafd17b45a147d060aa3924987cd6ef50dccf81b21cfcf4d716", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "20673f1ae3d0dd0a8bab2804d9713fb8e90364aced3d5a9c211a4800f4d42f0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "09b7f796f241e315e131b4889fd58e2c0722b1e8bab123bd7d6c974b98560008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "091a73c40a9015de8f789ede35eaf299ef74725f1700e34b569baab14a888d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c7525b0ed53d04c3dbaab7780c7bc7ca82934ef300c541052c25f4629c0d376e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "9f9dc98e2b18a26102e3e95bbd3a663faee5db53a0e82bdbb41902d11e1b126a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "73ce83524a2f61e4ec0ce33ddd2e8b75a0e0ab80132f35f12b2cb0c90d716b42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "8db2b145aa0313100c6aeaf4ad9497775f93e7e605d264378857e3fc3db55db7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "040a00f482f944b42f4dac9e373fe744c9eddc4d735b23b0f11354cf82ca210d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "02b0b45c2494b42ec2a27acb90dede469260518c150bc467aad852a22cd5fb27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "78fa52a94216c5e34ce3eb8a4226cf16e1e7b77eac987c15978acf6c812c3172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "43709a7e979c1b9c7ab626704a3e94fb9a6b3ee404f4077f3599ba2cc5cbc33d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "43fdec9952bef5df8a14306a3f50edeba315eb93dc5453598e33e6ecffbbc590", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "570be290aae6295ad23d4c47e5e08dbfa8e239d3b99f5e5af52548c5d68a5dc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "bf8f92df19d7a4096ee886d2494294a7edeb89b2ab7f6e5e53f18e03b0154c64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "43af33b84b0b238e5418e4c010323a615b213c5adae3b8b7c76e791ba2139a18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "b4513df707657836e7a3e321030bf20e5ec65572f059abb355a5f656a8015b15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "5de48a13550f9dfc1260295794410b2ec9eec7a99d9481e2d38823d4facb2cd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "4558d58eb4526f6bcbf87edb980a3a18cb84d5d38c91038c6623942847a95bca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "e630c58d9d3833ca5ae98d48a282acfaedfff9cefbc49c3b7017fa52d4265f1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "e11416bb8594b5d82185e5bab43d4da658dc4d7620b0cd9c113eec9e52ddc9ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "8a199a8fe3c172ed82647b79759677bc42974168395449604482eacf57500e98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "213cd22e9effbc09c9e04bfe0c2c5eb0591b62c093433a221dfc12eaed297c62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "ea126c876c471c8bff1b34436bd7686c4f3aefe5fe46cd0b3a33ee9c6b7c9d89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "505ab0e6a994791c2812238b6bbacc98862dd958423a01dfd50fca70cdaf2229", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "79e973e85173c400595fa6072cccdab09e34f9b7772912f943eb063d35a1c3da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "e3defe52fcb2fb512920c6ff5cb8a2d7e7761444934ed7712ecb02130d6ed437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "d5e84cec9bc2f2c2813173201b71f5a637d903af7a60327f25cc7ed16b992324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "47bda816d0bb930a43d26391cd2a697a6f2af0dc5297e9b3cef7e4e0a00d2706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "643cb74bd2f43d671ce4d934f7f2b927935a856b202d330c575f3d90db458fd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "868649b0913dc381b7661fa08cd4f85b4f0fee7215732f5e1d8c70176c87a609", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "3219a6e6edc4e47c40618fc208286a6955a33d884c00a277c2fc6f3bbe6f19f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "b2599dab583eebe4d7c580315ecb72e9965d74ab56cf98c2a8331009b560c291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "e5a30c27708ba651df37a366f293569d0c4945d7f69517a78ee56d6056c748b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "1ded0fa25f2d915b238f38cdf4428b4230456170ee9d094ef12c97712f1ab336", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "21289e3567f82857708ca2323c88a8d950d8990143bf75b9c315cbe7aeccc96b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "b706ff028f8153d8080ec7748e93d9bb8938dee61d61ff8047d88cf8b96568c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "2c69bb27921e877e6a26a4689cc259cbe7fad631de3275efbdcd4ac145e633be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "d7e9820f979f75d975f9f5b770c79e0f905cb83c71f3cd2699d1b52cad914472", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "d39f0cefaa50cbd82d9994d3c7f2da4e2adfd8e1e2b362d4aae89c1edafde074", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "5d0a6a58253f084d29072fb6be8e0e7bd2165243011fbe7b4cc7f1048161928d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "ea52949737c719305f2b9084d70f9a4a081b4016de594c73af2b5af205bc404e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "1c5b2f21ec19825c8117f79727a5c45a26a57c32fc0fa21d7245d3d37417633c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "bb0647a540bb9973112567883901d5b25581c5c2f7114ca7d8ed2ea44707a4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "feedc5186c7bfbc8d91b4785c28a6195763691a942c0aadd36651a20d78b5746", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "2e5e9b720cffa94590a8ed3e96babf048188deee8389285988c4cf357094af9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "e0c181b46c2a85ede67ace20159e806969d4b31fe963256adf75172248e7b52d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "2a7750f70ba670efef6cea2c6a1181feedee91487ab18ee638e17f6135a27f15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "bd04e1ffda9b72bd3944d08b2c020b5a906bad675adddcd8dac093a63046bac6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "614cb07ea9d9ecd1cd91f3dfa4282f59031b91b2ac019ec9d0e7317fc641a545", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "17b884e086e384c46088e7a772e382eb02a9a25d209f764d85db3780d17b86fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "5febe6a5429c5ca9e86fa7caa7d89260bdb5db3db4610fb53a56064b7e591ffd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "a64865b8b531d80c33fb8dde1b5932c8fbc6e51bfc23e2fa9acb782f5624b60a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "05196d6eb51e301217b7e9e0b21109e32878f34f48dcd339733810c732124b05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "3eaeba1b99e17c1573cd7dbc13c29f71910bdd8fb33ba6a4c33ead31adf243bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "503df8b3aada669fa4341e699159979f32949eff4dcf23e6e75a569e3cb5c9ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "97b5d4546555d5b54d6c6c7581818d181ad8e108548a0a54e4eb93e209d67bfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "cb2ef98fb074f3727e869f854d2d968f23dd949a1805953650cee6316babdb33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "c1824f33eeea862a5beb813d8ea4476c28da3b59ff8a4e7561c2122e487a3cd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f57c9021fbe1f059a84dcbb71874ba8c658c099a1add920bcfceccc0ea638c62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "1ff04ba73d06a507e99a3e08f3661a674030a8db9961b5f0fca19da5e5f70c57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f6aacfef9ee5337987b9b405cdd29145f1f849cc06d17993a873a18981412914", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b73dc40ad4bf18f40b052c890ebf8545b54c8c0e355870ce947e546b13fc4547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4713e175f4da7e314649a835552ef4c0c47c4db312e6ca1d39ab69ca6bd1b4e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "4a45aa3f81988b86b1ca55ed2d82799aaef219d3be13bf42ad0a36da779419bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "d17fa42e01de07fff89574668f61de9a2aaf9b40783052320407f7c29b67897c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9ea3393860a0ade55f2a799ce0446d5d4e023ea7a55cd120133c58d695594e66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "dda5e0f1eef55fdb586aebb44c268a7a1318837b15e165e44e314b70ca7807c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "850fb3eb9eea109e10d016c71b227fb68935783728d1a06bba2ba317f45132d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "29a3c48f1733cf65e2296c4869857102782a705165a300eec23b1d3c0187c0c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "05b3965fad37212be547cfc0d645b27a9cbb8742fb80e91f2fd7d14400643b4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4658d3b3d9dc3632da2228a4275d8b7a0405a14e865ecd990b175cad182bef23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b5588c8ee72104f0b916019e5dca82e19229cd2533ef348ca82734349cc4549a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "281cc0dbab024951ead9df00ba844662c601ca334fee73eaa9ce7c0d0cf430a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0b7c4340aacd9feb0e223f785aaee87499a3b61644c001b5444cf2bda4ec933a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "c56d23f80df36e2dfe8747c62444c5d14f66ee19313025dfbf8469c6fc9db345", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "ae7195cb98ebaf96b4549e9b6bfe2cc2e8b00d626c453602668c1de00113ad41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "29d5deddb0092519141b689670d926e2700c260c6a352778e6037d84175cd6bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b7c09dafd6bfa0db9b73b41a3a2f24555760bcaa28645331eada8eaeb15dada6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8208a6bab39f097809f59b7118d15cc0dba026a3fc9e5ab9e88ddf3eb0352c76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "3929a27b4a1bcaf8977c269f6dcd4f932c0e8bea68a9cfd6e3d70dc1c0404d4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "36dd1cd4817a76d614832b14a9e0436a950ad35a0c5156f47cb7ab47a6ef2f24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "a6212174c5c6644d7006f238c99e8a7e1f3d508a9ad376658c43f0e696accb74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2a937c31cd43048b63b0f233ee38e2425765fa0fe96942a56b5f3f71a7543ba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "255f89e643381cc54b61ab644115f361b79a9ff21d16b904a71d4c817faa0f70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "f20772f27e7ca2e8376af3f29b37317d2bf4f6223534acbf6b89e80094fcbd22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "de51cca2ccb97e505e298b6030b792e8f185df9f844f2ca6ebc67675d8157457", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "86d6d22a4cfb9c3103f428941526f596d3476f2ac97317be3ca4d6910811abfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "43385abb91755d44e1c79eb881202de2bb2c620d1ed92bc5457ffa73a7d80661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "9f6e82e4c07c45aae980a0bacbe365d2b8360bc44727f806902cc35487a16f80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "d9376e74e7a5c6b5a87352913538790b731d03d426bbb3a6cba22bbeeff04a70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "7a9a5915923a41b5bb497ecc7270aa1459a012afe2412049f403aa6a93073fdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "c580286f621d9aa16d9648f2598046d5bc2781851f216f082d6bc0ab8197f9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "eeb9f1cb6ec07305bf06b0207d34ad6a826366f69ecc3653debe8d281b35d682", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "03c61188a9ba761e893b68fa296604fd86c40c8fc38bb489ea6bbe6280e5ae82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "2b983168d7262bd936bdd000f7e10b7b80a458a13a86d620dfad0e02974c8092", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "1cd503282b4896197ff420d09f8295fcf58102f7d4ebf8d5302f2e5caa8a75ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "55945f333b65c91675042c8328b0914369fc6c5e1523eef7bfe3867366566a63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "7ae5420c8a6fd53608dca319373bca218f39e6d9ad67b517d42f65105b9db6af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "72a2c22a30996c06d2ecfe290ddb4df89837152b2b41d9592157683052c36a45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "12c386a31ecdf4d049c2478eb3d6793e092ab3e52ae7d5efbb094b8a3538d24f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "17ae212ad1e309d6b6e1a886b502a37b2e9d0253682b0fe5855be3709ecda076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "44f3041f882b6d8ae3150a552e97da019f4b792d81df417cde86e09dd4158dff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "558964af4bfc11a2b44821374621457576b7a4e2d98f84913bb5a2e668eb419a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "4a5caa17513b22b07e6910ad7ec0e684493a3a317f5f67ebf75e972709696736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "cd6af9bee768654c9b63416a26d7195a3dd0c04a6ddc9fa26f2c890d6e648f17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "228901337844ea36441108d939e52dc89e856953d050dcac4b2804707f898743", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "84e48658d2730b668d08897d6d2a7d9f373607268e7d44bbf475efa9621389d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "cad87acfeb50b59db0e0f4f8def241524b4b82d7d15e46e41b8c62aca325746a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "cf126d047ef600e9c8d3ea304443934d00e0511f4bfba69394b64a9c1bcb0182", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "02915995c19ef253371a863d8ec7e0d9ec45a457635164e6631fc583247ddd07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "acc8aca1c9f726dafe5c95857771a591dbd5ff7e1a3bcb09fd8e61985b9e65e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "07b88bc575b8efb109199d57188760a9798d103c252eb02473f11c21b60908eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "b966fcab08ee830283af0558d74e0212161ec797da7dadca2ecec66ea7c12fb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "934adafc8bf487dfcffecaa0c279bbb690442fb4ae84bb3f4f662469410c99fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "767d2b54779ef08134abfb462c280a6f5b35d9c6256daf1004cfbdb363d032a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "7ce7615baae53af81c36e197c585e4ff40a292d18732e09601b794db80ce3d15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "bd6dd5adce93f4fc81fcd854fe9c76d10dd889b9afdf8b53d04fd684aca24943", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "ed39da9972036346f9ab802856a5815d7ba4532799308dc543ec61903f35ff5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "397ea619ccb90962e3408a40950ac660e700a51b3e43ec9bfc5c4e036960470c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "f0f1c3f2dd8dec3796a407a850d58b0557df174f6deb2468dc592f030c451f33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "31e0143e0c0c1526cd159ef1e044c3e96cf164ea3850077b34fd9bc49d64368c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "1d0e8ee3a1c2c48435a029fe3778a55b469f305d5e8325d958bd393124fed260", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "e87ea0ce6f9227e1a2ea94cfcc20ab20d7355e1bb99acd2234051e94b01a0e78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "4c7e7ba4f4ff9894f5a53dc6f1b5bf4231f4d4bb0c4ff249e72a23e35bf068f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "50623133d69a0d2e1706d33bd5e147098c4138e2e0bdeacd1c5efb954f5b4533", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "135ed08bf30f9a60e8611da335d54f557a7d66c37fc3b4a83447072fed15d475", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "19c62709081617aa80cb5ad7f2ad7f05f86cd44bfe1a710281db4d182c23823b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "00865490ea25e934fb20bd6dc4d8368c9e68d3946b258dd66d536dc0d8480ec1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "7cf00fb3c2ab60ce259e248f83e01c702667a5e3b6a432849854ab9b84a41436", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "3694213b7b8795afadcca52d4ab4c42de604bbbe83c78db766778361a0d3b691", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "0d6d51c345d0731150c78214a28a726fc87a6356135cd62ef434adf0c41a54f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "2d921268c839982fa813d345bec237d60def6545c5764ff6a2df8e9b19e7b7a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "7c58b2add06d79da2b4d2d66a17c0c57d2397f4f381803b99ed84d6d9bee8516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "34c5d7a81b8a5eb8b6f5f0d21876060a8459eb128dae369426f6247ae993b184", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "20c2eadb7fdefa288a34f9a3da96fc15e57ca1039c5a5884d081845a60e03160", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "22fd12ef6a78286f14d0e50859581b723da556a1fcc32c9c8449b8a4725363bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "eb19bcafc4685fdf5d25ac6f1292d0d8268590180d42341826fd80bb7396162d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "9c4d3e3e79bfc984549d2a70aa479758872c9de2dd26db767bfb96a3fc7fb20f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "eaecf1e881a98885462721dc8187e9325c3a9f639824d237b3089dffd0aef281", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "94b78106fa40193513d7b46d5e4a08c1458f627f448186436ba629d753a86bd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "04d54e71dfd877305155a2d676b52ed165bed80930870d86bed7f9074e4c1673", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "bce78d401d02145a56b3436c05e7157ae8c76bf06a8c3be7cc2837b898dbd0f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e96c7c00e1433790e90701c0fa430e4624dd91d0d99820a12657367bc0cbb0b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "2a4879af62b964646be756c59aefcec6b7c914be45590cacc27d21808e8c02b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "770b17759ac32255da3714f5184758df9c5b8a6ac8c51ca95c018573e6717188", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "4e232de93a2b4f4324c2c1550e2bbba7a6ff24d275f67f68df9963a4a4d64eb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a72cf2ff906eeaa8039d428cc46a51bd478937594a04800b060ffdabfb560ffc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "66fc46ebe59b4fa1dd69281750168fca36cdfce39b76fe479ce48f59d5215835", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "5ace87bf270159083e82f9fc0befdf3228d5b633ec69d125953fd46c0c105c4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "f4d7e04168290862b792e431f52205555df39a04229b9ac84c8fca9222017224", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "8f488a8380952e0b181e3c49cb80dd5a396c10049f6da76c0850416631c8f041", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "f7acefb9ff7a4bcc6e378e47c66a530fefe517588ce8f251c1bf81a48f4add3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "b4fb1ea814d6a310c7c2eab65bcfaf906cda4479fc74cfe5b206d1e5bb8dee85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "16a51a03a2ebf5d1e3aa237b310cc8298e47ae4843d0ad27a6705c06a868457a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "f2e97a0b3333b5fa2a0ec0182ee1d5c5b9aaf087a55db02eb925e8388759fe50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "d27efc61f377e2837748b072a0df3415daed3ffdb2acb177211330b0b570be0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c1fa0218ffbfd64653a2467d2c0fb491f56c3e617388dc7e5db446d269a2bc33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "135003421f91d8a1ebbc4e4f9644295eca57ee294060914ca6bed97531fa9f11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e594e617ed53d0e0effb8f6a921e1b8242b4da05fb0d628efed4f6b9ebb01212", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "0384bc8c6ffd95e3c1d34ea1bfe54d33abcb69d28e50ba28d0d623cccf5e804b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "3652b0a5b13efc820ae16fecdd24eab4f0a1ce1e6bec861513a1738162a8ea5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "93fde52146928fece408f51b84d6e3e08acd40a8c314f8930f8bf832186ffbbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7947cb41b83b72f2fe05caa2f326504ca4e81f002172b6c12c4acd6f31eb1558", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "40c4c67c4844534cd863e5af2b0c94d8824bdc8dc6caae1c805ece97a27a5273", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "ffa2ca3f65dd9e0c2c6525eecd5425c0a6adba5dc3876639076bb0254674e8b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "c02da1342d2f7896cf4325a42d82f8fc7e5e0bb34ced85ef4ede16d0afc8f80d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4c62d03c30314ee3bf5d3dee88ab33b7a21c11d625657b5ef80efd78da279afc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "9d395269b6ca50d00efdd67d43566ca29b81ad213b7598aabd4a1105c12ec382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "7a083433ae3b4dc744bcf7a33e8d306d8abb2696fb8bd3fcd8fb5dca5498b8ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "719df9f2a8e81322770f25e6c6eae7bf051a8f1bb21375cfe08fdfdf210f6a1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "8f66dd5441b3ded423b6ae8619e4c0808164700de5f47291081484b15b62f52a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c0b36011c7397bee36211c96206569233bc22cd6da14ac91233d634d6e0bce84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "9c9cf15342a8ae46775e9b261a8055963cb547547a232ec43b8950ef0a2d3ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "263b27df183252105199847f23980dec79b68d4a3f9b8ac20991a0f088a1a0bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "06ee7494115f69a5245679209c07d4c24c613b4697650fdced9c7327b23bf296", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "2c05549b3342d6864255e63f8cceaa4d359b93fa88d570d0141a88a6f27b6c2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "54e8a7238f5e88fb9e0e6b8d70581318ebcfed7be19633683d4fdc819eee1668", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "41a0609dffb5174327385ef7ed04782f06498e66ba66ed90d1cf5146cd702cf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f53064ec897a86fd998f465cbf6e937dddc0d90664d06432f8e8b8ea12ffdd15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2f0d9a2a357b449da78c9b2b291ff8afa15d77d90863b717018a6f079df46d38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "eab77e9f3b4dd6b078ca41733b5a2b7c2f2834a07bfcf35286000663f3032b3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a7de291af75f1ddcaa22d4a1eb627469624f3e97c9e92d7dff6aedec2d0f3107", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "b3d24c324006d0db056eb1b0566bd4af1bf6132e143964a4d337d50fb8187831", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f1210520019c48cb5fb4681583db411ba428ff96acf71067a35a787ffc7e20ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6e441d0bd2ad4754ebffbd471c6a442b2f0ebfa030388edc1c8b653b78b5847f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3df2978f6da0c6992943847c0914ed15e3138c8e180c7a336a28a01ab90446d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b915fe432b65a39585b9295912ac43046f4bab0fd3ca6bd41156ea92235e8823", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a6c266b883dc51c7cad886b1b3aaaec20507e663ea58150032b01c56e5e809ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "7dd8f6961dfd7b5b5e4b74cbcf85f60e1a19ea1d0022e3f9e09010ccf9861a49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "428809ddea0666930ead5b78c550dbaf18b720ec3003641ba1ea800efbdf298d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "ab6d3d4cc9484d59426d7be1569372e837e86e6ab760071020dcd387e2393b9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "289122d48b1a162ad9a5634d648ad3ae9a92d3a44bde0609f5d05ff6182f7dbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "9386bb6add9078308a3fcd68d959cacf87267a2b8edede1c170481de171fe2b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "0c55c61f71fb7fee3c9c55846ad30c01a2b54abb6f084d60452108152f6be88b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "6bf559583e79a4f29c953c4b3a35aa5a4127f222b4093d10c45a50217d1b0107", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "59b2ef9ae9e549967f2473f4fbf434b2febb680f52251dccb1819cfbb1dd8fe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "a9885a374ed690c7a89b12fd31b8db23ba6e11e79e28b81d91bf5202cc90d07d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "5223164d6b9ebaabf51f9beadde5f306b5a3904443e9f936b4c9cdde696b540a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "2b3062ad1be9f357acaf49461f5a2a73a430be11520582e6f7d003f79ed1024d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "14da97069320833f66de37f31227f0fef9adce4775ebf9de57ef98ea01b9a2c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "8c3104d771c6a0ddc0c5bd73917ef4d1b8ba7a1941e7cca2515cc7217c9386de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "7806a6fb8644ce3341627aff100a3afddf744198b48d208585791ed8df84e1e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "bde4e81bfb25f3711f8054e3423162fdf06726d4d2a4b290e0696400e854b805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "396f34f9c042dbf5dc69a11d1ffcd6dd503ef708c8769ee55446a92d0f62f7a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "d04d5351f13c3e201463dc3cfb6def56a555ea67528688432faf2d2354908924", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "733126e802c51d0e2b74bf5d2bca299592648ffeb4ebac9fc73a24c12c391697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "4b88fae8eb7f9743bb84187d91d079c2d36ccf1e679c71fe9cde98e53fadba60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "8ee178aca80b0d2d447a08213c1efeec4558856730b786a029f8b93da82064f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "62dc5050a8c491eb7b19d2aad1e271986317358bef2c26d815f291e4189f34b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "24a833914bdb72fcdb8bef36f8fab7cae967b78721acce02e09e2da69ff6a071", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "cf35fea935499aeceea1ceaabc32248dbef39990452ab1ff3d55dfdbb4650ab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "6e9a5f3124ce856acc5e2ebf1364b7705438f1a7c40b67190daf87d2a7144a35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "3739a7963a43f448cd003dbf945acf36056eaa4100acdce7902e365d9401b701", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "de60113d9e1fd5f518af821fbcf7d749d9ce9019596c2411e3192825943aec51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "b6a8803d17dab7664ab29fae3dd268cb50f5f765cadff87162b5406274c2358f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "c434e1162e29c20ac25f5a04f871866852421b5531fb4bc6324d8b6120df024d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "672899871d0e33e602e9ee6eefb02a64a8169e2567b91fae21b0aef5f3e73c2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "64b67e6208a840c888a51b6d88a5dcedfdcd12aaec73a708f7d31911f887dde0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "4bf6e611b0f160e135aa3354077f8797aac4fa86b7bb5c99c0d4a89a722f6708", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "cda09f7e178a3aa57f476a73d623864f09659b3e6063f3c9693e5d0a9eebb4a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "50903383efc18cc1c57602ad7f9aa3ee0f6a4de7f922e038b8ae6fa8ba8c2b17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "b0a45f5cf647e6c418f64ad5b6f26271782a3b9232933e00640386731a4223da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "c85ae1f6be534fe8236cbfa9dbd32e46e601fb3cdc84ce458989d5c77816b8ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "561ffbc2c66b1607778ee3724a9615fa3b843cb9022d19830861d6a48c5b943c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "14e4295fd1ecdbf526315121ff8687b0a2bfd5708b1746e2909b4279f7cb1459", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "3cad883f823f2d0134a082469dbf29b0c67c4a766f227f069dd471be0092e40a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "8b852c4deea5e6ef12663d8634ca46bd80a1116c41bd354878ce37b3b168329c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "bcc3c2478417c3d7c008437a04091c856c05a55439512a6c908b0d5982850f86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "945d0d36523b94c079fc00e990ff79d0cff6a764a836a42861f74f5d21da8f27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "26163b0cffc48f91eea1a681b89d143b776c809c4aea2d45a55b602b4502739f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "02b13b8c1d9a1f80ee3c5b6ecf41be2735225d39dc07f20b4c33d1e3d883076c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "86085c20afce1e385ea573c25ba509168a5bed4097a4f2656d2e66eb5d75e986", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "bb70ba3e89b12936301f7327581c71c100f042b976c753ce19ff374a7b6872a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "2857432807d35e2ee5de8db15c78c88acee85256b9c326bb6e9c06b624f4a05a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "d8f47b6cca4ef5c7133ac8e48ce1aadb9b1dc1e2f223175926727fb995bb5689", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "04bf74881675f9733aecb399b8ef7116f15bf0d6192d4ecc292da08f62664df7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "8a2c63d020dac89d583948308e0accb955580c22064a6452c25551924d8c1631", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "146f95c70420b9e747ea5de575099b0831c5c4e44752953b4f5583d3818a4d9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* 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 */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "aa04ab4cd0aa8c69d52589d0e63f6d432d231703eb5552f7e0504e56bd3d3067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b9c50659204a6134058f84cca6d671231afad16782890e0c3ed68a59e19f5081", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1180f350115ff186e62ea6c08e47d8c4313517ebe77aeddf720d5166dc5f87c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6ca9f941edc55f466da0b6d9934281afde8cdea5be7eac9e0411c044253531d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "88591026a86e63d5679c37a38baf4b0e93cdc7250b85271d89b9f68558c5fdf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "ba39cf8c1f35f59153d05481a6e7deb46e2308bf159c3885563d52418b58d44b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2f65ba34bf6d85abd8f5475ed677005b75a4db98de43c8620417539cbd4e5059", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1a80667900e227f32cb0323852d4143a83713f65c4bbb33eb6c0c1c40fc2cbac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "bb80fe95bdd919b031612500bb4855e9b0737d203381f2f7ee952afa371c956d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8941630f38bd2efafe54f37bb3cc65a24946c4aedecfc21b17a13e05232d2bcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "22d59a14a34f439aba22166fd54012a6489087bd4becfaf0c498109808d7fb5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b15322ad2ca0db2f2239c7855eed83f75c1ff1cd60c72c2680d1c602ae00b944", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "7d510c1ad5a7b51465c7400d2a82cb62cadbc319626c86aa01a4bc059d697313", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e5221a957c030f8e9c810c338a1d0355ae021bb06e3c50fe07d24b3080c3a57f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "af40e8291ef5f678a3fe920b20b0bed4c5322745068bae35f7dc544b468a68fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3f0f9c57588d9dce4b09e8450db39ce027b3be4956d867e6029e9207d3446166", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "dd45612a13ab3ce2d424308f599d747c33982983124d43c5e95449679d08975c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "959de3cf342af375ecf9538f5e4904d6e323652b7509f420c783d5eef25efeeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "43865347e1fce478acc63bd425a2e3015bff4fc472d58e5d11c493482b17f86d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "7f51abff96de2181e4f8b160acf36d8ace7ae1c92c4dc3f6a327b7bff29892fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "b9ce3ca5956cafea1b7893f72cc672d34f6193f055a484fb3523e230713b973f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6d6b270540c3bfc69724f6a2cd744e024bc8ba90d9a0cc979aef584acc3e3cf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "b84854c051de68e322f95c49f25b679e3e31947b07e89f53aa33aae1e946f0a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "2c7c3a876a1d30e0910eadf6af140e60a36f8d8830f1e9d2e36464743b435950", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "c233e1812a199ed9165314348b2ff6e3b1c92bd7e3f27410d8f3261db337f3de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "acc3b4cd236e1264d05830567b51baba6a38a13f5cd8b1aa2f0fb1d02629955d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7b11d1009a48c37150f6d908c3b831e91293fc2bda5259608b99cb6a0de324d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a28aaf2e0d04eb8af6548d7e1a3d22cc4a1d531291b89c8bb9ecd5b9f1d318d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "5d0e70110182ba0f97fed869d1ce804d1c56efef29954dc5c31e29d9276022f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "f31a107b8491618ed6e6fed28677328502d1ea30f45ac8c39f28ab316255d6f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "b238f27c93f58a800c519ed17007a9c2f48024a28b0439bc3cd631c6b49ebd2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "391ac8c1ff5426fe7e1ec4342b95eb7e23dc5bf831bc3c1f8c4c5fad9317922f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "8ce4237e4eb26c34d1327d5495d2078154d797e705ec9dafd6db218dec186723", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "85de1a15c927a37304b2e439b1c8731feeb29a48f3e2799d2d92091bdcfa8f25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "0d7442dae0cf1b36a83f8b1e4a227ce8b931f2a414161aed72868c2301859357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "5ac2a926b0fd8a7c9bf4eaa59aa92ee177c275d6bbecd1c3c7ddb4e40c855899", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6641e131f83ab6f23f2667af9abcad8738e37db6e5b2356d1db8d66d7b18c2fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "96f75c100d2bb83b381b9f91baeafdc68ee1ffd1e93d7d2b50ad8431177610b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b9e1d40721ce3ba509d7aad07fd9801926e7995ef618f77d568ce39dd3feaf78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a3350e7bb8ea18c4a72276b7e03793c59669c089eac51a72fd9cab8a2758676f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "53d53b2f7df9f45d66b51ccf1bb5346b2e1f05ee8695498da773259e675cd2da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5b8f03eaf2b9198876a019514349a5dc37888aa63b22e60e73a6fc7cd86e77f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3dbf57f5affc1ad93bb825a98c75a9047eec7ae8e4e38a329757f2e80e04607f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0ec1b2c154dbaa8f3b091dc9596fa2498120eecae578512333c67397c5412e33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "73ca33bfb76b07aaa200147c07a039d73a52bcddf2b494df32afbd14a1903c4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2eb94c7f1632214978c76222969bbd20519302a9b713e527a164ed0d941ba866", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "08912862222b09ece028ec91b20ce9c2782cc2e2df870b190a37ed06f968473b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "cbcd14196200ab4441879be5feecb6c772a582487fe890ec749b194ecad3cbd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "40f9611c80cec185117834f17830a633472dcba6a2a5f31825fc68589b148dce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f45b1c1ed172561896ba01a80f6441610907c5f98e267b05d0b5825f1cedae99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "951d57295571bbe46358a591d6dd337516eb96996e73e5f263a98788741c68f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a7f6dd402fd070b75550469eb7e775cbc80d3dd1756456e62191351c259bee0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3e5a7bff4b095a9cb321e120181657fe6cd920f537dc4204e69b59230d4bc23c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "7c2e37808e11abef7ce785e292cf086f61b2979ddcffd82196097ca532a622ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2493e2bac91c72dc82de6447030b98001414183dddc339db8545872e9e7f9a61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5f6277f2c0fab3a5822cca1cdd334bbb6a4d70f87276032ad6584a340ab98200", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0c4151dd8033394d02161d10b86610ee081b071c07ca1d6c777859cc14ec7604", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "bb640969293caca30eb38569526ccf5e8d52e696108ca02b3d0dd25645887fe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "19bbf2b11e5a1bf9e8e39ada9935008411900b7dee208c9093ad1de5937dd609", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d224b5c839b911060361d6bd7bf1b5418cd229b4877875e931509ac3c954ee35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "bd268baf7afaa3223b5190874b9f80556646b2dd351a266fe76065381d8525cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "17302a016c76821d27b9002c156de5b013d382d1ba3c0b71014eb2f849763888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "d50dce6e27bf9088277deb41c263197c8e90640c09438ed50cc381ff6f62778f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "6587474b4ea4dfa28eddb95aaf5b375320e08c98e19f8422f7bc39656b677b56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e3f64eec46fd04939f453387cf885d3ab9025e106aaff1f2e1980c78940c7aeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "03c4122efade885969425238a8caff787bde923b1ff71d5101670409de83f439", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "96200e9971934e6ac7bc08e82a421af8acabd08cfb909d4076336ed09e0858b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "772c9d1c883a3f210a60c41dc006dd70e7de9ca27726a077ecddf0d0726ff4b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ce9cc3e12a9ec40f19e3ee4cad3c9bcd8cd07f3833420b607941aead42c3b492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "51209cde28b6de34251517a552207fa5a94f823a04d0950c44d2630c419340c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "04ce93e54f03c2745661873bd9d57a03ef27cf32177999ef2242fada61fd4adf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "02b50c01f20ebe4aa15e55a31db82ccfa7e61bebad53447cc5f3fbc8d8c38d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "1d4e09e2acc5f2d9d7e07abbb27d1ae11a81fcbc954dbe950a21277da5216c78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "353308ed580b5b599dca28ae55a8b319f73a0c27168efdbfaa4a1d65f9c73b37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "3519d9b83b22cdca78895fc67202127ec23a09624f0cda390fd8a02877f934a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "16263042b19749de08873e5d21bd99812b3d4e6a90f6795c3874db0f1e2a5b38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "4ddad2930506c9bfed699f44df69abb0053f674c53924e2fdeabf5c944cab6a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f57b6e2db6acfdca02beeccc564ad536bc2f265fe6f8e3c169b94f06c304e0f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "d430cfd250f2b64ec56b85bc085ed40c3cc2d85bf606846f4c2555d57b85c143", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "61ae3b6e30463923f209d6e2cf79f6108b37f71bbd8cd2841f83a06f2393ba4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "253ac1c7c7132cf11c9149ffc8b5a134c5764432d50f21be0b1d2fd8bed121db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f7ed439c676ad5459a002c72d0723ea2a8bda7f85644e1ff38ab2f7bf09f5ff6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d1f2c44ecc5e33a8e2728393d9ee6b169aa4dc9ef0b1b019aff5e8957fe6f357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8a48b0fe244a4e788d640089b5a0f4304b3c90088bfdde0ee027fb34513bd6b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f73466a522cf3a61d1c0f1ea9163794c71d83961714f8c307567c3d0fb04989e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4c2d75f0773df665810fc7b00a81476ea5ad65a6a50b99239405179e2428147a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4ba14df44055a4c71106cfbb92b5278845a8543f474313da9382472c909829e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "646bdf11de6f05122b78bc277cef52953501c785d631fd24c68c8b0b48e58949", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e59d9cd3c6e5ed490e47b56eefcac1b9042a5118e4f17e5782e1687c86412f89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bc21374de7b0c38b11acc3a4eadd3e59a37e93d021a38bb30adcdceee23dc013", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "49c88cf58544d61b9e24ca004ab251e7e0b23af5c4d11030529ae3e15cf9e9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b820ff500749b6efbd704e9e22a38377bc2d7ff8d45430f9435496fc2cae21f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "126c8c388304b70fbf902cd3956ddb21e0eb7e463137f13494a1103e5ec6080c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "13c24e9d581bc6418bb1097e7073cfdc335924ce798ec8d7b9d6c21c754d39df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "673ef41b16499b14c8c27673981ed0d52ba264844ce769ba7eb859dfdde8aee7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f6b99b4022a9a81a9d386fa562425cb059595a6654be7cbd4d1f0ed9f357596c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "113372034e14a6cbea3b8fbf2ca51c14879cfdc2521fa804fabd8fb007b225b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "094de5dc47a009fe9eec7cc87ddad1e7bda09b2a500dbaa570888639fd42c499", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4bfcfa07c94ec1a7ec6e4ae98c70cbf575bda8db546795c7d61a3ad89e377d2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "68bd8675c5f1253cda6be15a64153749a66ab6abcc4c38795c596a2a20f08558", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f635aa9d6aaa5359991a5712faa599d31545bdb114fc1a0ed724a0352e787b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c690dfcc8ce269f8af173028e51c9647af4042530767a1c1a37d4c0fee1e5db9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ed1fa4f85ef51a24c558f42214b8c603dad0d995fe58f6496dbd60873b1c12e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0140780fdc9954faeb7ae7336877abd611e4f9a5cf092794c43f5a6ee20f46ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6f9e609c30bd43c80c2be92679ebc2253d88e3596dc79460ebd7b7c98f7ccdef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "50648120d0f82b580ac6bd4ed45a17ae6c4055d91cdc22f7bd6c6214c9fe213f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6278fae7beccbc0c12ba87374f4d1a7b85fa8e82f20be4bb2f85274402476193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "da5975a72ac81212600b95f8996b11cc319177e483bf86875bd05b62877bf075", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "199127eb8c45cac8c2d71883c2227cbebcb08ff85bc27c28e55ba3f732331e82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "685a8d45856a9703f86020b6d50efbcdb6913061bc00fd9e5258e6ca7aa78aed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f1f1abeddcd9fa3594a6b90817dd1e27b3f1c1f975ff16ed06cb3a305f8a01e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "80d9fa5f5d0df0296ccd1bec124e8efffcab1ed8a221745e3e8a26e221ad75c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "164b1e18f4dc4bcb32c9c2bf8526f176e9c8e0b35af6fb3f51d2f83ae8392e1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ed902847246812c8e2e0415d8b0a32843bed8fb3dfcf6b15cbeb3ce333258017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c633fbd0f098ecd9a7bbe7a9f976b53981d22c1cdf5d8f34ade7dec967a7280b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "95e5aa5b6178bc4a1d63f3c0774da4b45fd0127a29b05eae1f4d42fdd0671ecc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "36c75a0b6eb601084fb47a3d9e85da615926a7a8236c24db928044d4343a4dfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5bece6dc8098723b3106d813a695e290e3f8d80a4a94e139d269a680481c7d59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "146cf6b3271381a9c483bb5da8bef45dede542d7c2b36a5951668ca7eb3d03f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8dcca68e97666c00cb8091b590788a99ba12e36744309d5eae7a59e9c8f26d5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1bc0ca21e588cd60e3498abe1e17179f46df5c6d595c2e487a013babc29e485f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "493ba4670b43b3972ad3096a9babb05039fa5f2d078c707ccd42decc4f56d9ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "88425283efae2b0029081ab47bc3292a10c6e0b00fd48ed4ab1d4afb0f4ea892", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9c61eadb327807e176bef729147b12ecb4553c1028d05248211186681bd64da7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bd7303d87fef8412e56ccb3ec6061ac799d5d6b85d87d4878e2fc7003ee4612d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1053a6bdfc0632c275c0ac219b52ced3e8134ed85bf5a93fe6270c289d0f638e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2e6f5679e6c4c76f2cb8febb1bbdc2686e05383c8840f2087d855471230502c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e618ea44523e5fdc7c38bb5fa29344c3280c1d47ff7ad1841b976d0c24ec382a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5f459505247418dec997eb96b6fcb46ed62a6021849ff9f0ba6081c69f7c5495", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "57f5bc8ddc9efeb3b02ff508cb9241e6f8d72394ce3de7928a37bcc4f375a62a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4d48e4284537951c1e80419e3fdc498775503ab5778a74a37d999922f778c527", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "192479c3e08b574290ff5173e08b07eeec1f7e07f012ee80341b5faa69682269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a3ed430794af7f345f63d64484b029d52c624a9a8713d1b99e5395a66653dbc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "880bdf5824e5416fada35f075d29c3299b4955cc260fbefcf66c936546e7ce97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cdf39a2e64345996b4a42f0b607d94068b15ab12a7eec51707dcdc824196c73f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9f357e8279c732ba45c98b2c5467dbef5cbbd5cf995137a3361eee32c9f64457", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "31f6466391461b443bd43076cb0ab59cdfa33a0576d7ef01614c7cbc11e9018b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2fde588a072d997a59da9ddf1d21d1bfdc47f8460428d38ce8836452cf43f2ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a603e310b49af9adf15f84b39ac0022d12da7b0e4deea49ddfbcb5373aac2ddc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "39397f32cf3e4a3e5a40992ee8db5583989b589788acd45957c4972635d9e0c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ddffbb18c5b7ef9bbcf2a4ac457759ebdf92ef41cf9a28ac00e2f6718c69fdda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b4779642548c15707c94369d38fd169ed41646015cb4034f0ecb8efc823f73a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4c527d4017c1e7bd11f7a3656c079a79b2eec62ac44cbb1ce08236e173f45052", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4163861ef92c3c0661a9b250ddaa6ce30e76188769e4cbc6ae31ad0a1357acb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7d5cdca3c8e61e4c82f9c691c7b5a5806da181cfe701dda41e34857a48365e17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3e03c52140976629cf1fd6c8869ff2819548b9ab3bb1ebb2209b9312a59e50d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "99a46b038e4ffd4f5d281548702443624aa3d65ad4b4f89e80b37779d706beed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4896019f8929768d299083a10e6acc41a469d7b5264c25c13aeeeeb9c633bce7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ee5f488a0a72b8531c5f97a8b8b611ad50044ef63b9a32948d28d3a450371740", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "53753201a4d3f70189f034c43d187bddbcdc1543580ec5287c99b249cf6dab60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2fb4eac3151e2f4eb8120620c1cb7ae2588e2c6b341413dcc58c868324c989f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "47d886930fe1e87b68060da421ae71a097a8a412b19c182172702a3ee62b9a91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e41642ca0158bbc7f4d3f7270f029810e135e646cf08c9c39a6b1f1b309a2d27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1eea435edbb9597f70256245e6d20a7c390af20465f78ce8b7e3312bb398bf0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "efc4abe9e16aa1005e913b141a05eb7b56602da9b8ec5adfc9ddc98e396d2ad8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c76e600ae25c98598749f82de2776eb569b4165fd79a3faf957a61cb6a13f210", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "febe97a1d1e7f0e61b341fc61c898f85e65b8642d746deb0bd2377dde31cf5b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "37bdb3ae30fe4b763f1ff809dfaaab72ed3908ddfcf18a289e7aaa1d47669e79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dafd0bf44bee6026eb1018eac8869dea17327c4c6f953abe8d700f33d19e6f02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5adcf6ed513e33b82eea9531a768307e296e5daf47584e767c17498eff1dbc43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "875c7d34412abf0795379cceb99240e22ccbd24c309b99997fc3ccc91ed42b9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8a9810d5716d3544cb5b03d255d856b3835be4e1515378009ef046f0acad6777", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4c2b6e3109ff224142c00a7321cdce868b3a762916a7c6600ba35dc3410aef29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "34652d237d642acce477dcd90997df76e60aea0a4e2d0c31a35a88acdb54ab5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1e66ab2798a09eeb15f97cb8a3c55842e92d1441eca3593ccb31021651eb697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bbbfd4e8d0bca604a507d97d6d084385e60a4c6e2f99e9d624ff4bb3520cabfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c75f9178178e263129227529646e371407f92d0e6c0ef9b0ee205493e665ebf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d678276c391f3e7f9469d8ca650647381875f3586c022e6b5a50384f6370c901", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8e40c12a94a459e616a1101e52bb5839a338609e621baf9922a49eb650d464c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "17a709442483553b0dc2d97450f351af048f4ae09c2fdf26f823d89bd1707561", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e08a5718ecc1cb0a38be3fd105fb51528b0e7e93eed268307ab7e87076dcf8d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "179cf00138cc114835eead83060f2b2ead2e16ce3b314dc5075010573f8f1873", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3519705bc311c985c0928c3db26e7d1ae0bd384b5a280a9da57bb51c6df41a14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "08cea94952de6c2977316c6eb31efe61e66c4ee710c1eefce156e3048784e52c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "287a82fe10423af93e62509ea556d76f8a6bd324c6168cc8ff0ed6c6e28c7fd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dcf4d87ad03ad50f0bf908752977a11a49944d16f69c61287d74d9b7d87a7bf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6f3b2564e38e688b3ef529d316a0af135085213ceb67867ea58c99eacff28176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e4e575faaf04f85ef1692b4cbfcc2a8edeb19477de72e9ac6e0dac726de80962", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "415cdb6125b21cf6268b85e7056a57ad2a5c3cfc4c6df103e61e96078304ad22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "11d99971da4f922ed93042f7326fbdc15e41c0f14b9d71a70500ce103072a93e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7955628edfa64bc70594ea166daa9aa4b1d5ecef63980f28e28b2bb3fd68acd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9ed564049ccd4e5b3207de801a287ee1898bb0c94c8675ad707db35a369590b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c408f20a70c321451260023593841e132bc1fbfb17ec49b30c5bba81e9f55ffb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ad0ff83970db1f1bb126d8043ed47942c9fd6cdf67c47ebde11bed816bf21066", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a887d3ac4248a9b0bc85cbf05e1aa78f473723f6e933d1db08a661e17d826838", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c5f540456d5c185248f90f453f41240bf9587fbaac8bdab942ca02a57b3d9104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e889564b7b6be1880b05ee75a29cf77b5b692b4203fa5381218bd3caf107b8a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "287d6c24dcc0dd607c22bb9772475b8a64a8ba5f8961748579a683807e9f0f47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "db2b3a2ec32225bf17691cfc959be8a1a90697903cf8be8525f454dc3ff53c66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0464bd08fa37d4663bca5a508ff5344c918d977d5ad9a4ed8dae1094826dea7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0bd46a116f0ea28932429c3d961d3757ec90bfe1cf196285e9ef29ed5412c201", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7e0d52810ecf56c68511df63d601a616d53444e8fc8b21d1fc0cf8b1c5623b3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8bb2a1f1e9e5d19cdbc552a07423f00cf40baa1657c37fda85aa198c8381febd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e9497241c5030f0aef5588c96d3176e8c0d6be84599a69aa1291d519946bba54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a5ecf1092018fc2bdcd74c0978fa8ecc75925ed527ff42a74b73cd407eb0fc79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "eb3428bbd42350b0dc61c442e5dc3ee754c04e3516354be525af4de2a7e15257", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a2be907cd729fc394298ffce1bf1c134cfb81c87e6b1897762946ac5cb4caf1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "46b555f1622fd3ac08e032e9b00196180ba8c62d40da4edef6d4d45110898d1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "86b88c0b0f026bcb8db2a3de7f8ec9502a4b6f25da711e2beda52d054e734d13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "618fcb2b3ff4bbf4293697fa08fca5d97a6138fa92cd450ba7b8031f92ee64f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6db16106fa0255b7e4339d061d836fc9cc98d2da4333694fbe3b3aaaf179a81c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "de2e277d5c283d9be0131314aa21548ddbf9fddcdeee8cc0788a46699ddc9c26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "95da8735befb75bb103e947f178815d82d8ef67bc7ec9506220af2dbb3942022", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "302aa7b79db8f85ffcf62d801a151d4e25e53e3245b43ddda2f709b5543df802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "df657fe860e222e53724b3ff0d72c29a43c7511435adad9e882ce92c0a48f3ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6db61c52fcceac05e4c31e6a9ebdedc3e11c850751dd0b894ba28e2361f1032c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0b62e468d39c60c1a40fa6e75ee64402d6f9d72d0c8373e77a0b52a072eb80e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "627d33ec35b6e0c88651508ed584b2c10c3d5fefb946b2b72cdf8e386a04906a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "66ce42254925f468b3a8802ecf8ab374b42852d70faf6eb3bfcd8ffc6baa68da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "593bf1942164e0afd91146da065e78ff916770311451f03ed945bccb9ac5291b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "12da5b097b3f172b5b15d3a62dc4eca84ddf423aa7c08d98ac4de2b5e2dd9582", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9d1092c89090fa80c4d4b3c86c2b531d2d8d3588dc0771739c1c2684644dd6d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "03fa8d5df33fdd28743a6fed9322d3a959a28fbd5a2c495fd20ba4eb2abcc191", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8af075d4483b3e9335ca20c320ade145bde71c7346f8c08f7d4c0bcc2b1a5f53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a5e8fd9b5bd9ae2f98e861ab4fb99909273648b5ee739899ce4f0548b2b6631c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c5af064723585bf2c855d7c2b9f0c0eddf654c95bb3d4790c29ac7223abf0120", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f5858907f5f0a1d521920427da1ea205c7f8f658093a5d2a8b4ed50443ae2768", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fdb95aad1dec1f9985e68665be810a147369c1ffa6ed481261414335ac62dbb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d5d3223d604520acb5cff3a362f8832962e0e0a79db8de7a42702ed4a0a612e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3b41e37b1f8a369e188e93852629f55d97afcf7ed7efc50bb6f45ba31d13b91a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8aee3d6d3044ab68d973427cbe08ff794447289da606644ba74398f177f9ca9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "63430c66ed36e71b1fa3641c24f1c705ed1d580213e6fef4692638241830a00d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "86f0f605330eb97d614602fe9f3c85195c852212aa629fbc8ff5ad3daaea5700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b660623e25c9b236bd42e18484d6efb9f5382c1914a1eb4ca2ac135242bec076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ae07418f7151a9e650725b1e8bf7884d158f034bb0c84fe97f9b7ccf242414f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1612d150983223a52c85069f885d1a8c76b6612ea57b9b44e49587f6699d2a22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c3a1f9ec813d426f37236a8c6007f9c4c7f4af3e3f53d52ddb68829ce5793646", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d92e591ff626f1d4a72e4575df8a34f8c99161baccd71cc791d18e05c2071c8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0feb9ef846a81e21972610f2dbcad8ffefdbc8286a0a625e694cd65b19500c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d0d79a4166e59271053dbe651b24fc756bb56916bd227f4f43454c5bfbc9bdd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8cd337c6edad43ef1aa54085f4e8f8eee80dd4415911459fd7018c811733db92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5dbfe268f38cb50a8722e5c9a5fb9d96998f6a2690d5977a4feed98989a798c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "97ca46af93bfa90f2ebb2f27fb514e6a4d64b706068978a915219464c2b944c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9a2765de819ba8df846f42eb613f33b2bf93831e6416063d9ba4f7c1452b11fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "582f22f9546d72bdd871842de1841fa215b8e2c3a1d6c462736f52e918bcbd5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "788f3f35a98f24f5363ca84d237d98cee24e2f3184f7da502e3182d65c63b283", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2e31bfe86084d0f5054f2f7660ac1755f9bac0baffce9e117a34916f6aa613ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a4ef8f34ecea1f3af9d8f895ce5d5cf9d005b59b874a2f86f9087e0caff447cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5a09a2e7c1b33db2ac25701c6868f1d75af028b4a9d5e05968f7aba541329b7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "96f2f86de2519754c25abb32df300d1d7e8dc978aeef8fbaa699afc5f74b72b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fb34702a2289ee8e106a6e9edf0663bee538bbc888d46997951dbfc2640f540f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b2ba5a441433a44a1974370563b968dd48871bd46e3a8f39e3ba2b3c6ef99fc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "84136e9acb2c46c00d97997db8832de9d9a56b5afe9cd14040ce80ec7141a06d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "aedc4a5836919c87a0da499304be7dfc81ae6db5564b5f3e66588a388eee088f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "eb62149984280c254039a7e7e8e6e617a7f664009ee95e2a9978700e6e7b50e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bec24ba9452c25163a1e6299180ca38431495c2076a68effd400012c295355f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ad9fecb3bc691c75b62e32a4d32adc3900bf6e5d13aa0def4b1f5fcb0792da0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "dedb4b75b1e07144d84910b1e4c3137b5c6f1330d6fd3edb5fc3487691f0cc82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "cebed70c5898d3fa7af7705f5e8aedf9b9ea9717fa8e97594f240bfd37eb9c71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2b8681d7dd841f74ab1ba1aa8e1511036b82fc55d48bde9f589b8b77f73e2de9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2bb45d3a6170e01481548a9f4e758ac9bab7182557b585d3f23891823923d642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1640e9124fb40bcc71b4bab786b597ce0d502b59ca8f1237a74113bca377f75a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c8d1c7a694f7659781639362cd9bda359b9bfd288e2e4595a5867ce7284bbe5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0629a2716685cda93c9507ea811a61c37ecf91319192278ed710086e16ec1ee7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8f638c596449fb7f150a26819c351c8c2842b47b4237ec358130eb689aa6ce86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1c894e6b8c22ca135b7435b2da3d33f66c954a8c5960ea25ea208187563043bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "af1f632fbbb7c3ba7f035479dcf47b4a60a0fb4f0eeddb1e60215e49d7842054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "720012cf0efed41fc6d5f37b3cc679e84e1784ed1415238137eff4a41de20387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "884eba4cb18666eb2d47e2871a67fe8910422db279af3365df0c47427b75083f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d8d01715c1b1f4681ab140539c033a1d228ad84495e14480082473d9800adee1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f2569ceec3604469471858f255ce5e655028ddd5420f75bd25522df5391db239", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b887bccc2961517de904b3b6666acf5b5b48d9895fa9729db61f0ffdf5224676", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1179bd0fbdd6cf11d41b5c55f1809699f3ae9d9b96a594f4aa3f4374af6c750d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "60b983ec78a035c1d2b5c62040bc7f9e080e82af84764c2de0011c8da56631a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "893d3eb27716e797a25eeda97fc4441ee04193edaf68e82cef1422f5388545e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8f08df543ecf123a84203504de9727f6d5ef60ac16b61d0e4b518c539d4de952", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c6b97b8a468c12b8bae35cd1a41d5322fbd01f4a2001c74279c3a5d1e5688352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d6dc5a28c0f4f1fcd2115a5216e0b556f5d302beb397d848e023ae03ebfb8150", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a7020a3828468fdf8cd34157616cf1f3960c085c15c1c882f5bed4f20a316f8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4f3720e17016f954c17f10d3daf8d62e51456187e18fd5cc27235b466b6712ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4e51f0ee1ce2789f3bbdc610d6b9eb5c96bde9609f5cf25e44f191857272e921", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e07b904d09eee39dc9d80adc37b29c9c05b3b8e6fa41ad87735c4ddc32baafdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fa97b4945d463fbf62fedc5600f7470af5d690c3157847e217b33d2739c60e35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "38cbd957e4393c4d1d5f6b26e5b2125ae4ff8488b5683724b57c499667699061", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "56456e1bde751f14975d954c1211c492615dcdef5374a35a3682d348b0c2257f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dc1c571f19d68fe434dc96fe488649903bdea4ad93b5a1eba8b246757647b290", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a1a53e0c7d5fb95bee58f6c1950d375c1e5bfe17845edb19d3e38240a84d4b08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "86b30b3221cc04236d93d9c896667cd9cf876374fefbdb3394b8669be3c89c8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "823140eed1b72c8f98804c7526d645ae82c0c029a43fbf3bebd4953dc0c12fe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "68c7564680e288930ed8d9234d0df8171cc093db1f9cbdf65a7de42810303f07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "18ebc607ccbaf70e6606705debd1b7b3dd7cba71ebafb6678154f1560b6dd855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "da760f640a6ac67c3c5e8c83af1e2097620b4dbe72eb6a65b9eebb3017740b48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f47e4fbc7f7edb23a0f9d24f32d96579a74876e12414ed14740cc849db29629d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c87fd7c40bbbc64545ffef06e168884af8540eaa9e6f93bb76b062e86e599365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5c4a2dd402b653a67c76cf272046c94f434ecfddb8380dc7aae63711d528d138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cfa71bfdb3eafe882ed3d257a88969779b662d8a9b3f544a795a36845cf7c74b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3f08df5c0d28bfdd8b87d06c12537af623f04072cea638fb4e2e09fd817f2caa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "012e96ddf13f4fd065e74c0d9ffaeefbbebb2a9652cbc663247b6bec3a45da74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dd21641ad5d30cbf0a0fbb31ad0643aee8a32fbc00e1b251d70adc1f72fdb76d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9ac0bc5e2d858f1542621d1f1faa1c9f9d52a246eb20f6b5190b490000f8afcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "516cfb50433fa4c8ff7dec6a617b6abd48dcb2ee08ad0ef042ce79126b8726bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7211acaef72faf1089ffdd0f106cdfeca72afcf0ab3eb2a20601d9f0bdbf2274", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "11aa5e60b62c4775b9bf6dfad2778f6a01f4c4ab65f5a8c189748b7bb587b022", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5cbe5913a1f7c9bd808bc73a1759a02b9da2f99b03f8da3868df97f6739cb8bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7915543fc95ff4e56a5b68d3519386d481b1f242ac76c086dcfdca6929f3ba52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "48d40754a2171e9cc2bab88074114abdeeeff22a7b9b6d154c80f7395989c385", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "347a4054fcc0f8edfccfb23eb8e8326f09435789dcf92c7b42e02d543dcaff4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b08a668007c28ebd14f14d892aac9702bba27c519dfa911f0ac5f9202cb79323", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9650cfd4da3dcd1d9d74c96dd9634fca4df10752954c42fc3f1f40d547fc4e19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2bf9bbd4fa72cc9b951fb1b1cc95fe32647a2e1c9c046f94b984042920a6e84d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2ecf535438867fe24f571af2829ca7b1f3397f2a11a70cbfd6a850d8e14f8c11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c702ab8b1ddb748490623462eb40c439c90d043b69871ceab2a242e01d81b5a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "15d80cca4a40d4c9f693604ce1b55ca048e9728137d357332b1e86819033a0dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "00ccc6595e876f198a4377eb1bd1da9408f5df9dcb2e97c98b1deec7ed3ad7b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2d535dfb8280e00e64e6cf366fbfd6e31bd65f71c0a7250507a323df55012816", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7c9077a524413b4fefebd9f855e71ec212f8e1b46b693e5d0e1a91e4a6f7d616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "baf8728e13f77bdda0abed20c5b5488acd5d00f0dbb6f95ba8d0daeaba51601d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "37e0c0b2dbcea8784658c5d692038755a9ab9760c1ebed01a9fa22d01e489956", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ed4ad85367b66e6ff261e09c19a5f1cadc3ae12db3db26ce45169b5004df4289", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2488889287e1de977de7b70e604a6f536afafd2d6ce652755c2d57fb06e4d791", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8017e28a3e3ceb83be7cdead88a557a2c5a05ff0f4a03c760880eaf4f7a61ddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "17af38959183c03aa91ac56e616457c63dc13222aa176a230f87b10dedc00e80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "92a43923f7ac11f2b42843a13b0f71ab0a9c7b4d1e8ab44feba75e484092b043", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "60dfafa1acc0c4e5ec9ffbc6f1b2f4b6bf65ee67bcef415b11a6ee287d6cdd31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f44ef6295723ed65933de81f8c041c00dd93866add370d024aa6d84dd7dc3b08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3c7fa3e41f361b6e54d8dbb78adbd5ac69d3bffdbeefc453f92296df889d531d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "33e77e3936158e6164a3f528913325a67f5ea1ccff6de0264b19185a3bf5638f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "49d600f68043a181829b77d65d969be85a6974a9919cde7fccd8af86c6c04335", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "36a54aea3e8472e2cc688f584f5a71a19a91b349fb84c649b5436e747e1e489a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a4f8c4bf3579b6c1c44371e3794873b8f03a1545e643fd2f170f1f686993f136", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1739e9518cce61d6459b0bde12820bfce6eb3521ab65c27a7a15db7876d506b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "efd778f2b8ee83c844502e3f3a90ae3b7c9c89339296edbfd09753a830598f45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "85b75a082d716ff087ff5c59f2b864fc0c1a9e54206401a41484389fdd2ba846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "98d1925809a1dccb4fcb949c249fabc48a5d1ec07f183bcb260c48486d226fec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9ac6c6ef38a7eb4c40eaa757d979ccca76dd1c578d309cf62ec24147cec31912", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2e9734b72956358925ec91ec6f5adcd821e9cbc19de0acf69c5d030068cc5f60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "28b2e4c5498cd0092dae446386f368aef791de61ae1676a9b4e3da955b0b133e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "54480ab10bdfb238a7d64fbee093f34f28ff974dcf19fe9da0a1c2aada7c1550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d902adfc0f093df1a9684005343db22b8552494d00cde77c8096eae2b4732111", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2abe1f54f8f35d00c1ce85e00b916cea500cba4d38cee1ccce65b88c6cd0a1d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "701782a906fe88a410a5454f5007fa51fcc79d21dee5bab3714e3ff1f55bb12b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "982e90465d6d24684e056aeeb183f1a4196bad7b8df0dcec687273d7e2498dbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c2810c57d7624f220d1656286dba0fefde84134439c1228b606bf161295bfc7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "12285f32e98164055437993f05dc41111731ca5dc5214fa2e64dc6271868bbc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f59b298f5b792205ad2066f1711ad38dc30b01f282dedd6d1d8bbad5eb0b7ecd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4d8193d70b64fd428b93510ef7b4e0a1e86e344c7b493c8173e857324fb0b068", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "904042fb87aad364bda1a7d8859bfe51e9d5398f856a8cbd9c42085611d38b5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e6bb03091f594e4e2903aea7f6d6c52fc0f1f1a6e85960c0351725a7750319ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d3adff7541ddcfd70feb432a332e6cbb2ec1cd2f9d2de9dc70b2af6c5018c9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ba4add3bf4db73871a93c909b9f05fed6de153f7695fe33d9cf6a043cd308d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ad99a386a4348519515db9010aaa7d171973ba79885bed07e293611135163bab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5f77ae17a6b313a7bbb6099a29e7cdb5dd3fc6561e37190c66ee352d9e994e18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d6e78e0149b921b226315bcfeda6ff393949b09bfa320b10905344f4c3c24fe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a842dd6571d691517bf74495e7dcbaa6be3aafb5d2fc590533614120bb10b35d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "abed598010a746b0f8d91bc5f473bd08b33838ad4f02def1f6d4c3578d4e5020", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4bed1c690dd3fa1af6daad1b7a735fd502882706939f5c05d69c0a75aeed19f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d5c1885c37ea17738a8c9755800827bcd7540dc3e734b6b6c2452518166d8793", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7697f37b741d77b5b145859965de3e86d28397dc156f382f1e6734192aae9b45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d2191203f52b26187896f6443583d923e2f9a125639980f082e039ba40f0f9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e5e5211871bed1ca6573b7731cd8d6517f5439e289d26b7550427803650479a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fc2db42a8f8f9c4254b12575c6a5e51e0b5892b1e94fe4d2d90199a369334482", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3482b081dc8729c3bf649b872b8b44b2519a0bbd0a9d4d485a769a885204bdf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8376635cab2a94a91cd62edb6f17255a7148ebe6e8e8a88aac913c508d5b70bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a87634179479d0ceb301d2c37301fc7f16f01ba91ad6ad1b3fcbf1adb7094a26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f58869527d42b9c6884ed4e8d482fb1f083f70274197bf9766616cc783ae069f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3fd95e7ff3333c96c18048a2cb2a7553510b0ed5c6eff300dd9ab24ff51bcf3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c6d7577570974008dd62bdab3b99e170edcaa0b6649f44a304b52b0387af8290", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b98211c5547c78ad27161bd730e7b2c03031e0e0357e957283948bb4bec54a6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d6af16f4c3604b014c8295e4e1ed23535d72f42e08580200e988bd129b889856", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b317d4d0073b910bbe93af90f712fc93b915d291c0752d8522e2a096a9d61bd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1ad55216932285c678a037506c2e960974455f5211580df6cc42d74868f06e8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7687b7261e7b9856b932e6d837d612574c7b4f018580f612d814bfb56aec864b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "85ee118d381ddbca121fa6ed508b092cde01c57e9aaa5384487046fcdce9e117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c9612ac0c860a633b3db1eee5434a6f4223e7330e52d6cb9ae8a595e5143f98d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9053147fc8b7ad955a7ba9efef2d80a9d91f0f2e0178bded6408995db4014a3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4916d41bc1bc2d06e94f0a50db2be9f83b3f4f4ee229ed9173aea1085579b5ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "43e7533ec187215dc232887bc4bc62922f2ae195f10f12f9d8326addb78f47e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "10e6207b8c8ea75a155a5b191f81b996b4f126397567e8f3d9e434dc76b69a6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "da3f8a99263eba1075866ed96da786a2b7c8bb7728f139e5a36b6ec80a81e17e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c3d1b6894ff5845bdb788fbb2554e61960062ad69a79a9597a9666807ed23139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1120bee4f79b76708fb86bdd749655ce1797bbd5f87d267f087ff37dcaa6ba33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5355c807aedc02904912a5c3222fedffb60114768f2eb9ad9a6199adf8e4de6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b573fb12d5b2b637a5069c30cc37d31fdb8db399bf1e3986d98f39239db4f63f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e50e0fcdfdc0384a54975fe476d74143607783109a05c2cbf6031e4f36a24c51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bfbbb9f16f983b37daf0a96a2254266e8e49ce70415bae0d3580a3151e8ff861", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "64ce5ed8c5cb52d4723e9cff922215335528899bd60d4e98ba69b136f382d9b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e7939a0161e0910801e59c6ce043fabb6c931764eea92e59856f93ad4ec495cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "02c68e74a69f95c3ceb4c04d9b28e5036c17fbf6be7aad14002ccd20564c0082", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5e8beeb8d832e097502b47cacbd2d2f4b6eb99e3355cc0324e0c7f9313773d6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "07cf4f2387b258340ed79221d34de9820ac6116ad2ba9209cd99d541ecee4b31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4dc522bffa5e7926c540d417a1677996f6b4f3ff0f19fcf1556c8766dc5248ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7c1061444d5355a27e3aa6082e43f6b7babdd3a2b78fdec05c0f41b8d0dd7e96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7366ae31e6287cd72e518fd6422157d9ef8993ed6c26496be9f3a95a2fa0ba41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4e612410118847f23688f9d4cc8923901155a3bec918b3898677b6dc0cb5782f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "95491df50f8a6d25fb5653614926df9a7db58c1ccf81c1bcfce563cbb4039b31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "21033fd0acafb143868c344b85002d4b38867a08bb4ee7b0fa1a27daf1e4413f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b00ed2b3b219518bcf7c06476145ffeae8aa370bec6c9187ca204570940558af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e3c8e6699f1bfa3a704148067efcc943b5d6ec8701bd0cf963c5e8b8e9bdc3e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e16dd50d7ed5cdd95a66ed014c5d2fdc91c62c74718f0898d0683f885f7cebc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f9c29dc1f84df1763be19c2034133aadf3f9ff76f94d220f20846730582e2256", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8e8e9036dc815a7143a1ad1c0596c7b872dc19aad819fce383c434dd7bbe69e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "41d77d135944c2de18c30dd515516b91a2582de5912c3f4dd2c679dfb1154c3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d2bd40dcd814309b86bbe6a1d16c88482c2b4d122440b89511b391526685d576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e61a28f472ff639956d512eaa6467ba912edbd4155933116cde12d25fb5dc22b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c823dddc8472b0ff32b338d15d14d62079366218ffb121006f0fec7176bd0ba6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "86cfed0c341a03978ff1c74e9afb68442e6cc1077e5355a11249642797c35da5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fa58ef51e0ade29f4c9f113ba11256f6058ffa04b9464ce32d9d2fc871705f00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "baa202a91f221427e43c320600078f512450f06eb0bc8268fffbb3028e92d2b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8991305e4d94708eed8918e9f4c7567c95e8ed99287eace5b0f80c6f171ba60f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3e4ff9f70f94376805d3d8863f25afcc5b821d036d225bbc527d2632f308dbf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "38901f7ad80bf5850f6c13f8fb4d6bd1e58036790c9d22d1af6aeec815bb4729", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "64d8f74eebda416fbe7190afc5024cf000b6f491af9855c9a0e2774ca0534bf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "671bd1c352bbd19f24c8dd68e61a169298b69e71d425627398633980debc2cef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "569f72447b21260fdfd055719ca60e9be6b0121fa7dc5393fc272d30357166b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "15d986897c077711b933f9e0f753accb23b08aef2a4038184c6e01a438ad9616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3011462a7469d88bc6887dea06c0a3800f43d8f83b116bb4dae1479337c84d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "38a7efb8e6ac00f4ec0b50860f7141d8868a4ffae1ce5271589d296cecda4c08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f5e20750e5bbf9c9cf29b15553f4d935028b26123ef3a91467da1ea9a93535d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ff2f71560e1c3e0a69e1d77f516989e2890397c6745d925a7c2b1c9624379328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a2318b82cb46b7d555164353abce873f1df4378f9fc84905b6f149a960ad88f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f097ad632c80ada9f6fe03e1ed1f6c41756fa43a591d58b33b2d65c291ae3e4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "35efb3b673c55d4be7d1d3f23460904a00f550ff3411f0194f910645792cb1b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "3aa15cb65a4181b5299af2eb345d5b8c3df237eebd58c049ea6020ba96be1d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "40a49df5621a4313d5797973f88b5ce831afca4a3a8b004b9f53b60820852f49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a5fe92179a101d88f81b4bd9aef5cee5256153108b14d95c44c2afa2edda099e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8e04f8b4b770cbff1e5e2090a2983e5fb68fad8002b00da65b1d0dff0162f409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "71d84f1db3bd7503edfcc1d0ff7a2e341bf0b2e43d30bc7d93d422fa3d8304e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0a6123e6cf378875310829b4380a61970b2644ef522ae3c0088bffb510acce3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c6c65d1289e59f9c7bce205579933094643d8b6872957ebdc40b29cdd212da43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "405959cb58af3723a699f8cb1d5505209479eef98af6016861a333ba52a997cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0da4a7f548f1af375edf2af763390907f67b15bb95c6a2b7ce5bec000b6fe67a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5440c4868693496e32b0bdcce1d0a122ff22f30313e4c7ed4ede5b898ac34380", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ae9ba9250509c01dc42507b01046329931e248378b317a12ba849ee3f50fa0c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ee4d880a2c39c5d0341daef3702055f8e1daca72992501dfa5af0be236073493", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "883d34de06398a5100f510f894216386f491c65a779876c6eeb3af25871cc77f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d86c03178477f96112ca1b230f7e5e348a35213aab4ae577c984b1727918e332", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ca164c3761bb34aae7c87d48b2fd4fcb9ef21dc5bcf3b67c21422bffc5490b22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "57beee2180767b98e1bbcdfe8e5ed193c3b150e232cddf08376ce9bcb95b130b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9e22b4d88349c2aa793abdd1f42c6823aa254cfd4732b56d8e015507d4d674da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9e973fa67d4ad6138333256ad6c2d947c7e03a92fef0161ac9329738bf396b2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a90a1dcf2c10322de98fb78dea534379cd0bd20a53fe140562d433fe825585ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "55decb3a92a0087a8caa8b00d9c28be6ef41a680f3b8d35ba2d74ebc9267e47d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "85fa638f543b50cc4047edc2bf202256d29e16aa33f3e209b950d70d4dee0931", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9cc6b01fa30e764c8ed102c35514843720b632083be7727a24294b95db6ba6b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "283f06f04ccb9444223e8b68364852a98a1fc84c558aa24e6542c4b7fa35e9da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "00dd6d26f28bd9d8c39c48c6b84240d41406f49ddb24a807aece3615655ec3d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "423ec50a4c77d9ee2489bda1e5b7d9b806f10607f7fa40fb49f2a37611208bbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "33b4beac9501b5f8c7e0beb093736c53fe82177a577f33344e8248befcf93060", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c4fa7ec9ea8a9987a3f50438d57b5611faadce22d810adb1605be10793be8f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e70f429b683f08fe0251feb521ad5d10709af7df3df96c369d224a7804556e3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "20c1e2617f1ad710cb1b9cb7472aff10caa5e6566660f81e16e9c743cd7cfab2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "149c441f1f2c8e8e46564d27ec174a743386e2175251e33eedc88740c22f65c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cdd670fbf628686c36afabd1591ddd2d408131c9b3fcbcb0c7696cc6edae4505", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "39556c5ead72e7a5f95f39e6c1b3b2a51e547bebcf8cc9efb819560d318c115a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9a0320cbd4afacb77f14bd4aa33059eb5e86a1fb69fe5e366b219464f9ba7a18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0f865250db104b69b2de90521c85df228506341ebd2da453fa01298e830c7078", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b151ebfe72aca70c9ab312798fed9559864f695ad3c730a6efe97da660b56a29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5450d46e81a64881c6395df0d33c3fd7af3b6ee518f3f1f7a0f15d5383d97401", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a9c372e2829d050cae657b06929697e4791d0592a24ea105f4f3adb67759545e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "86f80b754bf25f10531b4d48a351dc9399bbc08238d7501f49dd503feb836516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c2966ae352f003def896beac49de57589d8bfa579bcd3ecaf56957b98296c959", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2ffaecac995df18ece62cc25a0bfc99f9f3039c97907a0eda2b89b9b45300839", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1422240668f7b26515b6fea4ba5adacd59e2a8aaf6bdfc220dc241d3f11bc9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b2c4c642e43a6f564dc20c5c8f5e968f07cb1373713c680c6d8176059f7da2f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9eceaac0947b3fa461161a526a11ce90a067e993c67901960949f03f70341996", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8bbdad44fd4a40ce612e92cb98247d0074b6e05feb457b0795068501c7b315f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3a35b1431d6d20d2ea1a2b065f20dc9c5047979a7b120a85dfa44b369bbb065a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "997fc8926634829717875b19e74b133a66dd7a3f1a24af7ff44e41628245e036", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7b1f550caffaab09aeecaf2b136c75b77db99c72c607d3eafbcf93231bd87248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "16b0d4eaf8da0c1e1da883b510e955cdd750af8c515b4abb86a21358f2136615", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "cddccee3a22cc3bc56739b47a75aacfbed01cf61e765528f754cdd12964a6459", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2334e348881aa8f59ab7f14e1656bc841e1f414ca612175176fd1db2a3d6dc5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8b1a4ba4d998a1b0cb0686305de8c97d4300ba20c7e1c7f5d1c49eec2e3400e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7e629e5a88181c64490808812bb66c68639f8dc5bcbd4ef7bd5b8eb99d5e5df0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0abe0d9cd2ea748914e46b0535a3b59dcdd71311e77cc5ce0161b14cbec4b96e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6cfdeb286bf6881aded275b0537ab43c28ae4c381beb1feaaf3e9dd1095a7be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8ccc28c44c0220b86718840c4efe5b8da1eb2e76f4567ee97b267078ba17e341", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2d066260e6ea24279ab8a953c88b3ab0e206a584b12b95ef7d7e7cb6f9fac05a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b360209477fad8c9c50da017f94b6a69af16347b55c8908668da030f3c684c90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6e5b6a943533131af2241d92b3e6270cd7fe5b924ae183bf65134faa1eedb55a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6377484a75fae59c707ab9a2903d65854994dd01171bacb2ca0a3073b3d2b8b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fe8293be412bb3c6fd5084be4920795870994136ac3b14b888f4dafdcbe435bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "db12a807a78c64177c01d8605d4cbc625a0371a751af89ba133889923eca4d4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9a0d8305ba77b5bf4347e35e1b9d21c8e879b65edb014d5c5595dc3117e92307", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "afd10d1d33ff9e99039838a724a2bb0f78b2cb03f29630145946d92350b92d78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d7e7b289287f0a7e6793e6412876f12e0cee6934136f5afe0c61e73edfa73887", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d10c4ee54d46cdc687701ce9c020bfe06228117df139d328a8ea51690b08074b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "da1ea7c80a7e71fefc9c16b34e9ccd2508fa57c0df3e99feada18c4c7c656bc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "94ef9453983a470689d25d5bf8af9d4bfb0146e987ced60601fdc3c5fdcb9938", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c9591135f3713cd86b1f7c62e9d53f259452f572f9ae74ae8ed4fde124ec8d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "019ed1ff66c9acfe3daf4c479eebdcd874816280ab0692582f6211cd2a800b4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6237fa12fafbfe7f84c973297e78750883d962dfbd9c79124038c604afce3d1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4eb4f726da6283b404cdf98ffcc70f043b8ca9ca7619b1ce84943c42753bba9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a0da20d3b8a432126e43565e3fdccef26ff129a776831d83cbc71052f875d0e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2fb63b4627d88cb5d78723500cbffbf1ef727ccae27312a9cf2e5ae573503ec2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "996a70921601b6ed2051557baae6d8c5ffdf95296028ba3a35e3a94aa60090d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e8c8cbd36ed294f0858316bf742c74854adb1bde709304642d9a42ba73ecb678", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02e68bd3712a965a09b3563eda51103eb035862af41bd8a080467fcbb2a4cea9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "177756b6c2ee1b08f4b35e527c6f468be2b024a4898dbbaca0fc786baaaa5293", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "825b7524f8249d0a71ddc41589aee2cf4743271a0a9806afde672d15009c05bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3518a297d4f665d68c5412b4550e56f24401b8e07164e0730c4d8dec9f2abde9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "66d6e1f3f7b746aa5ec9320a9a0e33e41ab1030a25c88ba453ea23fedf80730c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "86f19cbd396e84a351e2cc94f09a970741ae13e60bf68a567b4e30147e04ecc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1ef45c581ee41190a3ebe29dd1acc11f464da67e4c1968fe91964705c78e4c8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "09af58fc89e767a947086b937da455b289913b7859116d2d4adb43003aca16a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7768eef519da83843fe97edf0758a56cda369bba78f72b2ed3e8d1ed71dae493", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "49247d0be4cd3abe9952412bd1b7a94538ad535b51f4fa7666716a81c43551a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "52861c47e6edb9d0e8748285146110c744710750677aba21f9d0a996343710f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "047776f19e38de70f907de58de73b987a3459dd6c14e0ecce660d77e856d7a74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf7ecfecbf399609ab7acb87bf41a8e13ad5317a3fd15286270b86c8ae1a3347", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6a892158f9a6265a49bd80a1c5503443e89b25a42c0742defeee01ffd93fb0fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "52a3fe033eb7d446ce98963deecea16964ffcfe9d3f2b773c05049ab1fde52aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7f5c0377c09d85bc50a4de1e557e0b6f56614a0e1a0fa2b9b817c47c11671d0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7d8dd9aa561b0e5b4708e46fd58b750903819f445c79b0df21a48c418f21bc20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "978dd2155932fc6a2b259205ea26b6a90fa562163ff6d5ae595d011ec9759278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "216f271d9dd817ef4f595df2434e9a668253120c4d1b79519595c838b269f43d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0713260f785f177139ab842a9326a0e3867df0e7da366223de14b714b92d8605", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "380f044f4df8bbfe23b2ac843b39b6ef79a222bf6b52d7f1fc2f83319757a96f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5f145377dc3da3a059f8f97b4ba08e284b8c248c9ecee30d06e1f9ffa3821c9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "79f1ffcb9d9c6d23899f032c1a0512280cb0494f69d6766e9d80322ad38541a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5ca4548ea16b62174d04a2b5b00899b2aa4a2b64d9b107e74caabfd4a0b99d40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "565fa5266fcae0f4c14b4efbf1e8b72370973ade0b6dfcfc1e22ac8ae601c090", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6f122089fc142b5976d50d7a4028bc42ec1654567e847ec465238b4044a4584e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e09f1ea930255d66e00f2c556958a0eb1f1aafe722666b6bcad6c9f8f5b7bdfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0cb0cb54e36221c42f8fe070c91907adb2be61ecc6f62a6bfea26c6e4c3914ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "29f7cf35e145dd1d8e7141c26c729e9d3b23eb095402a64075e43f08a459034b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "875998287ed8f9b05fef4147c9776cb074527979eb2e02be2b932fa0e3f34421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8ca199d5c9edc29a03f0f57a58785fef3c43c45efc396173210e23119a8a710c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b3b3a6c9b0c6f46f992e1617550d59b68be143c69e4e7766968dbafecc5f8e44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7f8331d891b1f3bb818815f0e97bd026fb80d81eb36252b5de3a0b44270ee9a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1420d132c6a3f1608163d4442096b74bee1fea8bc750c3b26e9c83a7a0106f48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0d7ba382ffc7b3351eb4b2d025d4c2ca63d7836e6b2ea857935e4cbf7ec7a34c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0f0673d6d231c871f453fbc7a1863ccd5f4e66ffb65c05ae46fccf72472a905a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d20b3fd9c76a0ce041843535c592fdf67c3bbbaae423657bda308f1b83ca623e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "93c4af00465801f5ab4e46743272f3af560793d25ed9ec95f34363ef366e121e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b39d74cc8bba19d61244e62ae2a9c08fd280ce9c8bc4dd7b98d63d9e28f030f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b824ef633163c314addac6f3fdcef45f869f7808a501bd2e8fdb254705571f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2b7034841cf54d8ec2fbaf32971d39b1bd2a259a857f7ae9b0bb8c6d9b70d4cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "069499e9d4f768c38c463bad5d3b2b214cf5b60d85ead063b24a6168d0fb60ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7e65d72fa1fa1b170d1b3cf755b747b15258f84fb8f5944f623f441c12ae204f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "89fa7bbb1ac8cc1ff4dd608cd6635853bd5abd7986815b1e53e6b625e9c09277", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c04b15588b54365877182eec81951a7f2afc8213b0fe2a660beef84158ea4add", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a4f83eb43d076cecdf3b2a8e50cc8aad02b1d2973722469528865e15f9d29d00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3738025ba74bfb13f3e00c69659a2856077998e2cac39a2ec9482533ae745199", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "db589174357eac86d5064a4f4b087e985d825fba2d556e62856832bac8a05d2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "30651582ec60b4f4d44a22b2899e827a4b5eb2a3d78f6039fd1a03da362650b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e0c15fb15bf072c6dc55a5e9a5d75d4d796dbf59635364d24914649a11eb9215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "42a03c8c43ebb54d7fa7ec3743f0000fb1a56558d899aca466c937e5f5b2b6da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e102041c70dd5c95503de15b1605a5b2cc627c21e08bdf63b7262f7f08bc7f86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "57370ab1977490331f1b4099d6c43871bc49c94f587047b602ce63b7fd258939", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8486b2698cd69b46afbee99ad582e266f3545278316ef1356a9fbebc4ec1cce7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a6645393f298c0a375f255d2d10ef49daf07058d14391875a01a509239110ef9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "57c3863e63753b99f85cbd3babd4758838e99f9f382bc9b1f2f6a2c536c16d2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b2bdb61eebdc41cb62986183a4affc2e5fe30567311b520710883e15e44fec47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8ce071d1be405d477b441a2e306ec7ebc867579c8c396fca9c1c0320554095d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "358d0c70af05bded4a2524849ac5f28705f989da7325fc3fec11c86aa431ee4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f6248d42748c3cb875ed8143e1de346d789f0eda8ef5d5eba3fcfe683018a1b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5b9aa8595fecf7c7fb2ed844cf4d85cfdad14f7ee2584a8d236eb95261c26dac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "07a4b97fdd8b6562342f87f28e0e9b8ca7736d103fc0d4e7bd352c632c28ea94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a455a720029f5832638316de55c4a2ba12f2135a657e642993523f9761c1f3b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4e10cea4270eac3d7f5ae32ee5513cdc96d974cb0127fc196fdb47bcf0cee904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "78902ebb55291e2978d489cf00474544f763dcfcc77cf0160db000ff37fadf6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "acca5cc31b661a1b27eb811ea1d7bffedbf27eaf1fdeb99bc593dc6977c5d7cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c5b5c67b3b634af3123c0f7edd89502101d0a86fdcc4a523fa960a10550539eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cd637414289c864fb1c3900d35125795d8654e22beb1b1c8c29b7d90c34c51f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2114f916d1ce63b8d7622fe575fe43165fb3dfeb208670203b12ac3e2ce6f857", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6b96e544a9e6c5760d2a14353ac93c6378656b33c1c9a3ef214532c214911887", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "054df0dd17ede5b9c9368939c02f5ad3e542213d5d2c04aaeb60e8a9ab53f4fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "14d104965ef87ea0c0678119abb15fd72e7a9372fa99b06a318a9d86cd06ddbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f1d86bb13a228e3e7b8d875dacf57ae1dc152a4a35476f07f9ecb7b6602c9797", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6c1c7e813cbba90def4247b28fc7d84b754390c5c27418cdf47cec9d1b3cf387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c8a05f83da109f688faff48db7f100d65e88987e67d9db56dc077cda39fee6f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6001176f911ef9066f8300bc8afb70b3e8b609c2c8f2f07e09c967682d609c9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bb578ca83b177e538c47e4a753318434e170ee2749e742441b9cab6daf2c2e90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d6abbba963c8d0ad5fae617702de769f32955a0eab9c3884b8b51897e4704115", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "29d2c08abb87f98db5bdb6b6cd40ce159f303c842958e26639690cacf656a1f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4e4e6129239a1b6d0e42682bd0fdea2ade9f1413fa8d29860f5370a26f3bc63d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "fbd5fef0a960774b721cf7859585b90ed0d3ea7643277f53de9fa7fc9cb5958e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "65089d18b5feaf4227bd896cfe939e861abdd42274970fb7d2a8a0783a9dcd2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "297f785f3dbb8def21fed2c79378f651e7dccf792a172a6c72afe34ed2d06b1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "368f082c01047a92b42e58140a988cff49280ca542cf34dbbfa0e15e197c0286", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4ae0b6a8dc9c7d36203524da9b15bbe52248ee9a3b5e1109795539169c7c1acb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5def7e4981510ed56e8445ba7cd721b15d4052ed00db951cbcaaff4135cfaf65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "969e61deba7433ec7c2b8db9e601a45a13dcc2dafbe6649e22276aea266a9b90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9ceb77fac4a423d84a67de0e2ae358d6823474b7f9062db63e60cf20a1678886", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d8564c894fd514ea612527b09511b89177655aa615dff73eedaf2227a9d28340", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5ba248e7361e7bda47002c26d56255d54447dba199f44d258c13a29c2dd57568", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9593ef00486b24a1f3601b733f0702374b9cb016d34767dab3718aec5a493749", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a73c6936c16c6a3dded7b65542b9756204db1ba916b34c0f17b5f4a283285f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c464a2379624fa454646461f3790ac76c8e794ad4bffe58cd8f1710c65b57df0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7c45c92d83a0c9f94d875c5b037632994e94f3732a33abe93e467b4efbef20f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c81c47353367a6f594e717fb603250a27dc3740c8dc878a6538aee329b136e1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b015b187f0fbd565b0e2a17fff57f1df9bf96c0eeecdd811a64ead9531c2ff22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8ef603bd428d5ea995fe0f10700d06f2d8981fa86cb29f0447224e9c25225673", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c3a7f0ba564bf6dd72f76b85335ef3b61af38afa32ae70afd581f77a5b2d10d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6938ebdbe1a2a57b628afef5ba2061e6932a9597eb443d6476e7619898e49ae6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e9f2d01d573f9bd1b732f382f7fd39abccaa2ac10fe47d605fe7ecb06456beee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "51ceedfcffc452519fe3fd8ac40b2da2f2467254e3317f572b70bee6028429fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "afc631d3687d17d089d0e1b60e3aa8be44fcd7843df23572ea38de2546b7ee07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "178ac048cefdd4d3dfffb5431c205298fcfd86879ae125e59dd74d9b10c02c7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c99fd14153b0f349ce426328455dcc8ca51c46aba7b06cad881c72ddce395f24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f24e189c1f26b7afd2c6a610ae92d80bda4487540f682d4b4c470fe1c98af0e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "12719522718d16eba4b9cd7938dcf7f53d3505fcc953ee03a50550aa9e416541", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "609cb706640f137483599bd8b98a58905dacf4709790a9215ce5a906ce9a279c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ae5405d677e34febac879362d3ac3062523e4f0604b74747f0dcf0b34d3397e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "71c34d47dadb926b96beadcdefb6da7653b77480cb60bf99929a9765a3aead9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e32cadedf063c658abda50aa793ca23ab58f65b37fac5915bab488e245f21e96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5fd76ffb080be5c66603f1b9e589594a3fe34385b0ab30825643d25bd7f28fed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "688f3cb217258e2d5855f165dec3cf0c903316ee394ced15b5a62e990cb4a42e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c63a1218bbc72c23dd12664479d53ec687cf6cd655d71b5b90e6c0774ae1e0e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "256de5d0a9e793ea5ccad2af3ad5d1dfe15abca900b41257355a806998aa1b7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4c51e4019b4d0fb3f5c7262ee5bb9b10e7cad15109afcf0cd0dcf45a0ca66441", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "eb9db44d5cdd905c86e6c348f8384954380afb6e912bfe367ec735873c8baa7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d123555d2357ea9474cf8d681da9a97e38f89d43c349a883a319f960b52cc027", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5f6fb712296b674223f54519dc070ba8b8293f466869708a53f6f70bf0dc1e8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "607bef200c9f19e515be4d2db6949b7703aeea435abb2667898ddde5ce43773f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "995270b95fcc59884ebb92f0d566441cb907bba5417d7db9e2b610874a79c9d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1928e75a86d50195db584aed1f0397159f88e333292f142878def25d7e271208", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bbafc71dc6b45f7021e22bff72632dfb6afb4592358c907d17ac24701c65c0a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ba13fda9324efa7f460e30a2942ba94ffdfb822eacee073a94fc36666d7d93a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1c03d9adc9a6acf42c09779ca4625b2e5b88193ac537db80b7e2676e762c99c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ba88d4e2efb85af0b4a222677ca947fd447e1555d63139b984a9040d3e461bc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1ba8052d0932483238037db34dc923bff6479eaf350b06fb8fd2ab1e7e15730b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f502d9fde4b3f2af072edbb8f69f7201236455dbc139f00970eac6c7a852967a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c3b2a9270adfd569bffd75c2c4d83383d1648dd572e73029421982a10326864f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ff61dee96986138d20734a2366860b64097a7742b48ee9a51a05a1374559278b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3ada8a28c9aff4d3045ace642b8a9f5909018d34f88cd265b57b9415e62a1094", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9fc980cd1634552e93e731e2576a1e4bdd8eaca9c2928c083888143f4fdabff8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bf6f8662274c2e6c8a7e64a91f81ca93a35d5dc6e73a90b3b2236d6d4a12c661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b9b76a58f731a660f7f372ce12d9e53b1cdfd867a752c58bee8145e1d60de9b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2c254f9970bcf00e49fc5b72ada6522383043afe2734a722ad5fd428b625dba1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "042f3435528e5ea8184e4bbdba00a3d57b74211ddaf555b8e57231005cc2e9dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9d62f26f2c106c282b623653e42ffc6dbbb2e1599c5418c058c20c0f86f6a054", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c56d3a1aa0c0b340c7ceb4617d1ba1d0770db5dcb85136dabcb0f7a9cf460560", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1206279019a61e6bf85d4041927e74381c6bb404bc3e310beecccfcdf2ac098f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "db112fcdfbac3cd6b0fbb3bb4774a39391f88a8719761bd3711cfb0110a2baf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f21d877cedd338bfd6677bdf3c82033364829fe4a4a5b014f460d6bee6c43d45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c52741fbf9b86b62c555da89d5f0e215d46c2b94fbb291a5490833c6379fbf98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9c689e89b512b3a9fdeca2c89a517f039a3c46b879e26741b22138902a2277a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3b282c0ed2b3c62269be04561e141cf5daf8cefe638dfb4dfb85a92525cbe4b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fc880736e8f6bfe842b4afe3d1fe7792d77bc5840b5d08b9afaf06330b28514b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9f31bf17bcc30d459c2fb750bd0c57233af11b371688264e8eabaefaa80d0b10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "319579b94a7c148d181df27dd3239601823f2f77ce4860b2d4f4129d867a48b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2d3c880e9bdb03ffce0342cff6ad766a28dda83eb0d304eb0c561e6e9fffb9e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6690f0127e2695d7d7176d26a553dfc9588fa99a5c82dca700af5cc013f84049", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ba20b70e2884a4091f230d183c6c0c2c2912789dd377f150fd966b3554abbdff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "68755463990fd2942281213d42622502ec282f3af354c0b1315036b0b72de69d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f08520d727f384c2f213a919b24ddd8959d1146e8df2dc5674e9ef409c72265d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "904793af51f359eb0cec4f04ee6278f5f38ffcaddd5895647177e0ef86c5f1f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c60a5bd81ba16c28a519632665269e6bb89729a313938481260179f968be9a43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cfec85fb5cbced06c4e8f3061452ace11fbf4099a63539af08c1c5e23cfb5352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d35af2870d751195c86a9ac86c657d890d8f540bf750c7bfa0d6594a9ac46474", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "688a30f6d17f95d2ebc425652ed74f5f9a34bc323054429b4040df1cfddb50ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e4a81fe644e69ec4a287bb5bb40430527927d9686be2e910756a62183870d3a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a8379853273ba911db60be1a7d2cc67e6b18712b61d522479d30250983a970de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fad95c68e675e8b5851b93f2b03c701dfe04c07496698e52775f51ea6ef248fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fba205d531dd735e05250488e6e05476379b5ea52698153c83548a89a46a7826", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "16b0bf0f8f8a3a680326b30bf4a8de60796d63e878599d8ad3d2fe94425e0f3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fed0f94971b0fb3ac1b76cd5d07608b414cdf23f06cdd74dce85d3f36a1d7937", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f95ee15c905b943deed23250b6a8a32e09abc791c6a2556e0d35d982b566c45b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "71bda83e5f505ba8ace1308f545127f98b4f7f56764937063563048d86a104b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0cb6470566e8bd667dcc087682faadb7dfa91ffb5ec036e09cf47f296c4cdbaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6dbd471b8ad6ab4b76aabcaec3a5fc44617904f560d80b729e0d7efbc88a90b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bf787b58134c94ba3561a85127d3ba9c5314de72876140f032337f584830e7f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f84cb1034fd28ade9544fdcbaaedd4a989a1303701cdbe673a920dc6fce6d376", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4ad1efa2034c9e389c5e3e262bd1cb883dc819d38ed39da088c339f96a2434b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "18647e0d8e74cc5aab1ab1f5911b533d537b4daa2a9906ef3829b7d72aa7ac7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e6239251f9f61b635aeb940f510437f00ab77ff1e4a1356deb739a922a49f4bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "49b924ea0b17ae0d489cd3a7a639e3fb05409559397984675031eff605cd1bef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "63ab7927ef5e44e9bde1f906e0cc1b1ac02d092e0b81345b11124cbed9f23788", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "778ca94e5f3c4f0657a5908421d2976af634adebf6a2c4fae5f22fafeb73277f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4f3e2e833a603084739f2d590685a5df50df1ed0530f291d68ed7a4446e25f6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "11d744ad5f7abae0d037fc29e460003f2567e3d58b1c909becf4e092ec468197", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2dc0ac98db77dce6628b3d8d3c20f24cf09f0e8e857d7dac16e54ef1f9c1905b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "073704bc823a8459414298f057ba2b0d7239c2d98bf34fb4d41a501cad2fd8ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e18b195ecece1fcb641767221caca0e572c71d7142ab51d8e8c6b398bc289eba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "86ae91f95dc2ee91858264fc60ed419b40516000f9a75f2124af4569c08e9698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "edc7c140d06a707574659860a932b78220eaec9a1fbd85a592cf530ecfbf103f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "24d503c74882f4e6c0988275ccbecc76b0d9942be2fd359fe70df2f62c6c8785", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "362121e61c0bb2746e48ecb914f8eb13decafda0267e251cdce04468d4f96234", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c4062e34139bdbca66ab788e55a209d0c00ab590919dab50e2075cd44af467a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "088fb35dc1214b0211f61a9e27834c78cc3b53af39e21fa0e15b53969ff891f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "21d0a8915016c98e24db1c3aa3238f97e3dc1e0e66c8b348eb7d5ed543410fd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ea874c784532477de760328cac74069f387e3d42f581216f2ce00706b8a860f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "19548d94f17b27548635b9fadab95293bc1685057771277a657a2e8d876c3e07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "497e97eeba7888548e19740d87ee622f72c8f4ad88bd16254a90063c97217892", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9d1973e8a5e2981411ebf7a610325ef95e35f03a49aec3b14bc39982680a5d14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8fa47c4f95b7ad394588cc0080df07025aaa0431907e263169b483654ddc94f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "989019b2feb5b1c39eb35821cc350ef66b5aa2f0b556b8120e4ee8ff6198301f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "492cab06a1f0f6fafffd08932d8a10d3c9940e60fbe4b387df5af90f4f504d11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d089467938c583777d2d84e1fd0f23606df913ebedea793de24d9fa740c9d886", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "57cda81ce45363b3a6262dee678788923aa5b5004bb3994e50f7a52775c83f6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "01d8104b2bc39e312c27de0fb3e09d581a300b84d6b2e06ecbfdf4c704ad9710", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "acc507dedc24200a1c665c493c5fa70158e6a2f735c6fb56333f1fe3929f8d02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fc1369fa895fb71fa03c77fb621f116169db83b756bc0431d4e00267d3f540d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "13c99a6de79dde1e0810ae9164e3e411479a78a7b18b06d98357b00f6ee1dfb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e3e2b1fca176129c92dcb6dc176cf206d203f59206e67edc18eeb645d50c73ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b6f71b50df7da794b5bb5c728bdf094cfaf99a85c062181f8bc3da8dfe892d7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "819d2a28a220937853da4ae5a5cdc52a49ce6380707d349c85642f5a0a6db84c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ab6280c2415cafd1f5c275dc6217bed2d91b4c4a0d9d074c258d10095aa034ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2662489598d830465036df3a867a05f3c866ee255d73f90c08e9f7cce5c32c23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c616f70d34ee988673839080ff98b325f1da031d8ed0f1f83100e25ef55d948f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5fd37169e92e1e5cab43083ebd37cbb94a549745023bd7d60e50b2beeb4b3e5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1a6a8747d83c37b39e5e75f3132214e2cdd9d66f25aef474c9a2fdfaaa8ed39f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "07fbae08e3a58d333c0cf5b9eb00cf472d8d5a7b0b85a48a59e8c42ad285aab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "037529cf3a16ec00943e464441351e271934fb7b2f820aa56c22573097597dd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1e44e0e4ec89a9ac9da8d5d5ee5157a3790a81558c286c851acb1d5d70aaf62d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7f62d537823311c193df18da8d280dd007992cc3b9540cd2ad5b7822a9b0879d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8132e52f6bb6f1e7244accb8a15eba95b363be2d8c41277e0b8f9ae010e2872b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "06a53b36a7e56637ef38a38851387a994fa4132be226c5a4b896dad977d61fa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e8d92cf7783db561df88ddc6f3065af2dd6c17667a16ce093c247fecf89a0be5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e1f7c0dcbe59b9e916b859db5b953e99e2b52f7653692e051714fd6cda3d146b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "74091aef63a645560d31eb3f8f316f227712c8f54459aef4fc63cbe58bca8f05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7ec3b8eb55d9f94bbb5e66efa2c9584fc7e04a2dfa6f22ecdb68c271bcccfff6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "75ec72818eea0564226052b20b10913c484734b007791ce2a90c4e479e409021", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "13e5fdc0a50dc68dacbf3600c29d490e3f84fa44370193ff125cd14ebb88e5ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8456f2487500815f42870188c1413d156fdcf6be3900acebf5fb3bcf62478450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7b146d97b6939a267c8b22f83b83518ac5610e00b7ba4b597967193e3e1325a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1d91ac719d5bc1f0d012754a1613d65a3b3bf442f63b2d2b39895479f51e213a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a5eafa235109c383527ba7f681e4562ce2b5179b155b676ec0681d545cd97404", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "065a0a4e3191c811e14bbf499eff1800d662be3ff208ae952c0fbf5e04f0b7de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2265bead701b135af6cc0fd76b39afa695ec99c3c2a75d54134d98cc2e0d893a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "de5db058f6265da1779967058ad5d8ab036aafc0d8e4eee8100a87fee0f51a8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7d8444a2053f1a8111777b38d085eeb448ec923d9da59b231ee751d493af117d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "60de9e07a470d9c65aa964c3b0db6880d913f5f0e7f010a16742256a37a39f65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ffd363a712142d005d27afe530e497b25681f47fb431d7d38597b4bcc77ed9e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "813bce453244795677a0f70853b45ce6c926b49e26409ec54fa1c997b4bb80f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c87572dc4c26f1e0a2b27dbf22e9310040f4be038b92d500f7f974bc828fe3cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ffb96069b0edea4b2a69d3d0b927f8c9b2412bea5491819c1c89b9cf576b11b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "16a4b274c2715776e21a465a7adfea1685c8cfcfc5c2d1485b87ea97b48568e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8085ad7eef898c718463d75e130803005702459d42797233b769b69b156669ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c66656494937db533eacbab3a5115ddee08d73f9139874341d562ed5f10b7cbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7069ecdc2d5b8f7cc6da79169726e659eabe94548b69ab51eb9f92ecc6b7d7cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e18c1d0c49f7d0b66d0ee29d85abe9f92ca51a36fab8f8f6e0868ee16901c53d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5dc1ad20edc1b48efdf8c61585d65e6691b803e6cc46bd5a1af83533c8fea8a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7e627c938144c14b430400404251858f91cb8b62c037ec557689d24c8ee1811d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e0848ca68681b0eee81b6ee2672872be25edcd4a3db241826acce84627caff5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "cdc0a3e98096efbdd92724a36d788c44d3429abd4521438c0a5b5b9e819891db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "66b1454a58031701e55e31dbcb5fd8fb9135b5974e7fb7276f1b5cebca21cf66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6be1133301dcdddc7beccdc0012f069c8486350cd7ba5bf989f37e964bffd4b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b57848d7f65b91d3fc21f52472c1a8cc3ea0ad3ea44dcbfe99e446da0ff1d610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "74f680aa336cb8dd815b0b8f8ed9b0e524059e4c45a952d6f62325777d517598", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7582a83948d975aad954ab0fc9c82854dd48d1bebde1b4eb7e427cdaeccf907a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0d3be95b5a06cee19902f289d78f0454b129d5221bb281d537f33fb6a88dcd22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6f9b599e7a302a3c4a0959a9e4e1affc698362124523b597317a4eb62ce4f0d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "09fcce24e86db5992dcbf45cfe0d6cc341f4cd10a2f8a5c378288ac25ddcd6ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "3ed7800099e9c0eabe34c7cebc09c7292d983d3bed4c4231d9932092cdffca34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "99c8cffa587d210814f2f41a23b429d0b9c458baeb36238b9bdf8dda6b597410", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7415f233e64fd3591be0246167d94368788a5aa2d7952be70a618709ec3834a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5e68f04eb9f9199261d90df86451202a38319384ee793dd96b977e4d9169aeb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "174c897d65e5941b98d8552eac2a1e9bdcec80f29873eb3c9e577f0f2fd33693", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fba1e6b3420c83327548179587916f9d2dbe8addaf7ab55846a5bbf9c0833ad3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "98e26df82f606d6fb1fcef4f2b1a3c709e14318cab3e1ce613314e59a7d8ecd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dd6705a1b3fdc146e5ab120296079e72f7f8fc782a4adee5c91e7f9e49d84841", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a0e5c977bdccf2ffe528d6daed477d160d11154702fab7c119aea2a667f2c9e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0ab5752c51d6b15e4451082c6a10c9ca76e0712dd56a3bc4db5e956ec5e94616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4042b3106aa1bfa52c4efa6dba8722f0bc1a4bf62cec9feb0d19a8d12f047cd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ed29237810e44f690c932042be91ed3ab897da5def770c8fb11d4ce3122d2ba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1f035fe48b0e5fa2bd4ef74c15f783c2b44b9e13725813a36ce158aeae286e04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e1c379d75f5f8909279c70bf31e685525a4cc80f6ccc9c7d762d146564d7e399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d11fd4305c2f812dfda83eb550dffd2d3e8bfb9d97c0453b0bf91bcfdff3e90b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "611e8047661b72a3a558a07c9b6bd8913f2e37ace460179b67e4de5de7d1bdf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b356b326ac861728ee0e058a582a0d53ec18a51c26f789e467eff35c3e3921ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e872d89434389f329f96a8ccf0e2a6de6fb481636f1752e0e7434ebcb2f564d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "81628e7b0e462d8c0cc512d635668bef1bd509cef08cd486c9c52297b4f588be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ff47cb71993da66c28cc46c29bd3ad01a2276eda7807bc6ea2b949bac65b92b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "cb9a59f2d776c978019d10b7dfce6d5fc3be8dfd4903374f8c799f8b5eb6de4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4337e5a44453feba28f3ad6efa6c6c6fc13e6c079a230f971715ef2431fad981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "497465aeb072a22a170e4a2cc0437daf8d0e41d828aff7ede988d24697551b07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "391f8e74facdd051808145621d0e088dff0dc836b09126f94bc2505f6e4cb70f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b36cad41c0b9005ffac141589649f99a8b94d15482158d8c9625d69dad81c549", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7a8362df2378aee9fb3710e2e3b4318b88a994927d9a4e8984cedeba950394c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b9a8314261924ad78a71d09ffd361bbdea637ac40a8c50753e5f9c0b56d6333c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "203e533cd75dfd80611353c2ce919ba6dddc360e255dfaf29f61f3c61f7aa0e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f2afc7025498f91b94f25607db78be2da02270016673386d2444fe4e6723f0c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "63ef8f47a467419af9cc9ab7adff7bc07c8eaea10d3f25564b76648de00b6b8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "616f97fe3017412a7a01d8bb0a098b7875c5aa532a1ed0fa554fa306472238d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ec4129842f58a7625e87d88831d4b2cf9fddd92081ec2108268e12c214759793", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8bce58fd530b68b7af8928f9844af820848c321236f73399ca070af55f015c38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9f986220a7b44d46674880c8f5b9ef13c50bf88880f3a9cb0ea6e48043105b4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0dd4d32670bec1eabc103ac9e8645846c453043fe8968388380ebe21b2ba1004", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2da1376f68c4bde0eb5faff7b6a42f33f969e104498dfff078053bcc05b6b763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2098b5079933232c0597fd70766cb9499ef0555ec7c6f1553363060ca4b32ac4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "86a651ad82ad67e24b3a7f858412ca3066ec1561a6ea9b31592ed61fd77dec05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "dff386477cb1b188aa385e289df5aa6deb60f3c9aeef0861a3b5c958285b7534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0c733a5bd5a5cb8ace024980fb441c053004f9cf0178aa5ed0853ce7e014e985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "afbdd59cdc93cd4e3cebc4aedfbe84c4eddbfefd525b52952c082f8fd20c5cbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f8b700235c103fcfec51ee3ee72102a12675a6d4118d3a3e6a0052147ddc8d4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "58a9a3427349c49f09d7eb873cf33068b8e81a6b312170dea5621ea280e36d29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1ac11e660e03e6cad6cd99562a97c5b3936e19f01828e2598907a7d18659a8f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c651a84baff0464e53ec8bc5da08288fdec6cf79218ad0b473f03aae727250ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7b40c37d446e8e01335660458426113f281416ff417ae783bbb4af7b526462ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "825100d937bec410ebb0ad73b7da564961e8640dd35ed55ede1971b7874c872a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "f6f201a9edfaaab413d1c36ec6c3bb68046e4a176679abcc9aef742f8456a1de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "7c8735ed4c474e7b37f90d53cde8c74afd87b2d84f90aada4ddeefe037d3b211", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "fdfde2b77e340fb89018549c2bea4e803910dde0f3d148d5171c890566a37364", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "eed3689890760475f7e69b8decf21c1862a7dfd090a53dd95432ed2e16c7f011", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "426a9042c6c0c57e2029d5d2ad271495343399cd857baf6a9200bb66d166efd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "0fde97c52ba598f37cbeb1582332c4c44cff4bc70e2d0ac8c4e32e978841eea3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "9f827377734e3d1c923611dba5cc330ffda646ae08c8dbad9044f47af8a48ab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "1747a5353230bf46164ccb2df07a1143a9b3d2d7017b4df7b581682500cd88ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "17b73508a267cd931220ac4b214b64eb5a5aa3815ecd748f3ef35a91803790f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "67ad512e230fc1b4fe98e28fc54a970e6061449dc2c9791a553d805d9fb8945e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "503c3020cfb56ad996387266a6300755e211e305e5aa62ce010011b7c92de8f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "2356e81fe44fd1123a9557d669fefdee72644a2dc7e1fca97dac59aaad2399b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "a9fb8a6104cbedd7014c651a9bc0362547af7511b34d6924dfeb9a3307357131", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "94876e2f838c53caa743155bbb7d2b4911af9826ac15dfdad59f3d9a42aa3c0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "62bc7c86b652065ba028382f55aab4bf131bd58c0f6e8b003ced3a337747e497", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "d6e5ab2c30e6d282a6442d3e4c2f6d81cca09a72551fd86ffe5f7c44e3ac53a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "c0e5bc9d33f37c456023bcd8a0332c1eca7bfb973ac4cfa9365183cc49312202", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "5da17a2ea97055d01166b3fc849d2cedbddd296db084e37fa50a707a1ec34809", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ce5d73948f0d90ff4fc2f21f12e5afbd69f312d981fd2584c319715e9ffaa546", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "17abd827b9ead55b4cd9a87507eb0e46bd88728e0418f737eb9cbef01620703b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "4959e5f4935c9c92c53cf5b331a4d01d189fd36f9d7e43ac5fd9a0924afa3d67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "4c13809076b29cdcbac92da719129b8f30f8ec7794056d75556788ece9456beb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "e7b842bf00d348699600da1f2cc78e05a324ad26e8300c30015cc5a997bc3434", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "6bc0903b566a7c8999bb381338ce074f823a6e58c5915e5c587380c556f00fed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "3f0385fe23ab58b0c3b241a8ccf15394ce1a562b635737e06618a183dfdba94c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "611688ea684c90d150a448959ad474089dd582f6f69cf3028376218550f771ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "b73035216c0349b05ba6564cdd05b1462342e6eb8c218d6ff516561ae44814a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "6900078cb72a345f256960ca1f234822b681e5cc271ef5f65617bdfdf73e2aca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "d05214e4ae597162a88fb04af63dbb94a158f94a5d5553fb7479a3aee53519ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "6ac65e0c86090e136d6ba4ec5960fca0b95ca254c991e540c4870549df9a0c94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "7052511e8b569bbddb3c5f5475cbdf7e2fa6a2ea15795399e5d049ef2c5514ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "2f248e315c362cabfc8de3c8f7c33958f87bad7a00f752206dbc7a76ff194042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "08ac6a7664596694934c66845ed23a0f937d9a50a701d9640193080909d72d1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "19bc4e6145e3734eb22a2e38be648b0f9fb7903b94384dc0e560e51a5058b86e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "d172d39a73181e8ec0d9863c0217a3679f82bf11a6cf31b6c239de4e9ec72c00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "9ba02bc68ad50ce89c46c84ba726d93d7b8f8f8704c158f4191ad5369b0044f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "24ed709a39793fc7d8257428fbad8910a3b1bf9202c5ad5a0be3591b5d4eaf1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "f632ac07cfb2e71e7ab131f8322d3da29ee4cf8389c8677ac7c334452e4a878a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "3e28cac517614b88e752a262afa5b2712b521a911745bdc8bebde98c3a4a1fa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "2b4cdd8bd1b438bd0007d71176e3888f7becfcd7550154071582467ec69fbbe4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "71369dc9e1d84d7b0079cea28d8c40dff50ad432c4602d5c25694d23edf7ea56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "8cec3574df243ce7070c58255ad116ae0aa5eb1effb084bcfcbabb6af745bf19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "8fd584ca7dea740784b004cfdf870d0e9f9c64eb3dbb70d60364cd8d0b2d4e8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "6dff652d8effcfa68989285355bf464e4bd2b89e7ebb33ff0c7d8eb5c934ee0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "4c88162594102abfb2bb04c09b3d4c22e6f0aa8425c40bd093144a07112d86e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "10005bb9b80434096ed22aba6e22b9499c20508fcdf274ea0f5d4ced81894d50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "4986814b2f337b69ab9d02c720cd02a90e35a8c199de742da22628547ba60930", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "f2ad9b818bcf7ef4e50a9e473717733a5140d26528d43381516bc12bd76612ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "283c056c2c866869a0e464ddebf69d6ce02ebbb4ef75713df900c521840c153d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "3dff4cd84de98ac207e7ddfd18218fab1fc3ef487206e5e722d5b22b62474129", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "de40b5a7dfc2168846b9c38da876c5fde987eb49a398c4ad017a8fc52ca65686", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "7245dcde6c8b0b196e0484452e75f9d1b93863e4f4f790b62b11fdf006310399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "a12cde5aab7bae4462ddf4a964814f1642e00ec6cbc4eb638978958365d676d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "416f552ed5daf21416b19f7320b9d260d25e10627b3e6bd1768a142affbf329f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "3c3e64129ec0eba5762160e4477cdc7aead96e6aa034fd40f030f57359c1b11a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "cc799d941a32a64698742326a38f8dbe0b2f2dfb762aa2f7cc19adc00c812034", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ed1951230f3eabec3554d8528552f97363512261a84921efb6c8ce8eb625dcb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "0cd6f4cebd605e3586b0f81a4fd52a91ea40bd90f591a8950ed79d40d4628a2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "def79a51cec13294b6d38462b79d00d47781b87e029b9d60730713dc296c5df6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "319b9f2c6545c1415230048a5d160517388f43762f94d523ba923deffa7c8d3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "4d0e09f0b89b2ffd9f44f0766af8b4d3ce511dff4957aa2e5bfbd13d358d8d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "1136706908a440d5bb09960bcc676fe3bfcb0114b335e751699912bff5d3a0b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "6fab98e3c175def2c8d8cc175538b9293038ef7950d8a63ea6a8b0a52bdee8f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "115553847d47a932b6329f23d8f14bb88e9e38d87357e5d7b0e7b5ecd98eac91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3d54f88d3add1e04cdced159da41621cc16016670d857fab62e1cdf89a4836b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c6450fcf932884c69c60c9bf83355d944f2c7b3ea2fbf9970dcb012e5ba74de1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "84039b7efd1b9376400086be62802a45445e45c9fed5e544ccf17d7e735cb732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "74fc07c8c13281d1de432076081d76503a7ebdf46a517b883539a42897899e23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "fa7b683a29f132dbb5203c6766e8ea02db54a653cf649bfbb2355131d5dd7a69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "ccdf80540978494b6181e85975ce2abfbd951307ede16395e0a4cc5f0df71818", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "3dfcadcd18f9d0c43f8c9448caf17110abf105fac3e24e844d57ade8030d1e49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "30dc5029dde8a5dc4831462ffddc64e862365d5f92cdf5662c3fab3e9d172f70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "481d3887bf6ea7775a1eba4f425cd6363719d4323aaa602132207d6511b1d61f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "9e6b7c13e4fe21f68add831df7f5f78b821a0e720bfe0b66df068b262a1dfb1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "cbf51ba75cdfda6b23a823c5fec4cc0053de4a1e7b9d1820da58080d89c47300", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* 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::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "8c0f4f8c54ab62afaa6409623aa34c48d9d8686c60ffc5616ef81499a67050c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "e95f2a9caae23ae51f1791dd68dbea99d06e579b46808e5ab6329bacb818c7ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c5d40a7064f0acee6671ebff40f556fcda58ea3a3eef5235407fe568a1b3a4d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "005f0669eb3e14f43eeb9779bfee342cbce6dbfe5012b7fbe9d6abf7c94dc9b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "fb1edb21f433e6497fb53de45e4c34120d97cedbba7fdfc55834bc96451d6d45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f80a8010f5dc5587ccf819b58bcb18d3055440b89e3d08fdc746d41babf767b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "05150849ba2bf8ece6a7911b0a8affd7fc8cf88a10d59e58d311299499638f70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fda5d981c565c6c26d2cf765a48de8f7be7c1310b27246c859e2c0c67afc5d42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4bb55f23c223b29c782395f6c53daada5b9ecdd81ee76af7a44c1b121ea55657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "942bc03ac043258f67f71151cd7e23784947bf78680131b178bff6d65276752c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4d8a37ea2c0962835e263fec66e799903b779b9904957f986af53a9ce2cda484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "faf6db9d1567271b7d789c8dc6efd0e7b0c45bbc7c47289821d0b07b716328e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "47531e6d0ed0f975c8271869880f1c7ee47937bef9ba0332bf2f368a73a3e684", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "41de901d700eacba5d13d81ef50a206d6c199b12921682672bb553922e8ef0f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "ca3097fc3a3a3aef4e53cfa53ea3afbfdcbd646aa33079bd4a0cd8efbd928c67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "0802077db8726e23a7d4c88e32042549aabac3e42435dd8f7fc82fbe70f36c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "62baac27b1a862cf68ab30c365f8b0f2669e2e9513237ec027347853218888a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "a592a60cff1364637bec67f6636ef5c4fe17ba69365529e89a86cba127a5d27e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "4cefb25595d28e21e358ede0e9c5cfdd0509ea2ec812e38feeb6d7e0098e8459", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "bc803955088c3848f8f753281ebe91525d052d13df6a11067c9193d0f271158e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "a7e63f2bb2c9a8f88618dab9a03c5a6f02ac4cba4c7add16584518443ec247ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fd424d2c8ac39081e9c36dd89afbb12f6739d6cb4a2cbc7ed91101eb46d2d7cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "08000c52af69d17d31b2c5202dbefaeb3f43959c77f77828cd2b6e5cb8c16e0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "93f07f5908e1ed951befc0a351a90c5b65e12e7bcd7e6ac2532883c5cf0f5932", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d8d0632348fd0bb03769f28e2187014699ed0610769b142aa4aebff06b88cd57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "08d5277ca29105808ce3aa6b08a6d9ae932c718843ae47c067ac85425a28f5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4466c116dde12d89df392a403b1878f23de7a5cc0ad4e05d4721ff52eb97dba0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1d5ed1be15402ad21cd8c7a9da2c54eb2422f0a4b1283e141cd5c9ac5e3b6215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0d8768d108965d843764e480298d103e214cf9895df75ffd10bea23cbb14e664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "24bcc0d77373507b41bc26484488680527b57c8a91a9d3e463364d3e7379ecd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "5019a1ab9680bcf2a8ecb76d38f300ef41f8fea264d2bd4f0cc978c4d8fc7531", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "6e111a5b5f457459451308a17771ead5ccc7230933393e5215e082188f8bc97b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "7605ba36365d0e87236622412315e7435ed57522fa99f333ea2bd286e4a47c75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0e2eb6f8c2b21a94e2e5b7cee71d50bc9804e894e2f648d916e260c478bbac15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "0dab7cfe6bade976bdf8f024a15e03c1f9d20f1901cc572133fb458c62c96254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 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::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fa3860eb0eb38c9ac87d0c7f526ac089f64c76b57cd429cbaa8a206c15a9c9aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "8a7db1754eea0e0639fd0d78b8edbf9cf2db917f57fc0e81b072c6f152f454b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* 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::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "5d7e7c6d5bce33b19c3b32111b8c02f1b42b4b36669843071d6ac731b39b38f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "02ac86d0f2d914b8b3a5e45c65f74e51b10ee299fb2c8d6f535ed1b3931c128e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "53b8c87ea18ba5b3da3d561a736907818fce7a571046052a61b1c7631f5cd3ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8f561b2596075d14d3d56a36452a2b670b621179fa5da31efd02ba524d5891f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "91870e952c166942c48f054fd5218e27eade6c71497a61fd5f87e88af44b05f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "30eb106d0c62cd154fe03c963d3c7d7d4798e043e97eb6baca8fd54b3b531fc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "6c979970217524997f05e794b1cfe6bfd411870f1c83a40db1771bc12b99de1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "899a19fb6caabeb905c01a3bd74d4294043017a0b85b4205223a2e4a68fe00ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "28c56186fa98ed652a7b03820794be7fc7ce0ef4cff5635ae136bee7c0acec50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "72080ef14b94b8a2b947ed21e00a4bc237b2010c61b674b6dfc971309e7952c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8bf8babd44c5ec95d668f7f62f26d0c302df01e49d10e1bb94b56242701c5b63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "16651f81f2ceee31787be042c21ce5544182b0a1166cb73446469c268d018564", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf2f954963c702c5350dda3c8ae0fbf5f868c83e8cf87ab1a734723326a65b1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b45d589f18ded99876f57abcfef141d443603d8b3f44e8e201a0d626f4f05f8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d0ef58cc41136f944fe1bceca6789dd5f31ddc75a908f918fbcae258e03085d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4abf5c52554968c08b79a65e34812b006b088f6b43a8089aa3224c3bc051a38a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "48a0f4da692fc480f279a4d023e31526af44f93bb2d93d7bccee034a595b11b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2d62eca439dd0525d41db58884b24912b80ff89012742fdbf273371fc5bdc226", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cb1a7965a1e160410d823d012410b024ff1b4bc05bf479928a79785272805121", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fba8d1058920c2d05e4736975926c9bcaebf1e2883889497e05e7fcc60b5de9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5e8fb42954c72ee2bd0a38f67c6765e6e840810862b003d0057fddcf708aff21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "59f648f9ebe72b38ff4632533c65ea0770d384ff1407c5d0d8f7fd842aca8a67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8447e29fe3082cf0bbe255ff79db0578312c1a7796d1a258f0ca21db6420f085", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a2763ccf406a6514c31b0dc678ad9d1bc23e9316af08f785690621f0263039ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b9d0a02e9ed660a9823d9e1f3b8d2853357e710bef4a7a9fa2c8b226d27b1a27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "85e77e8c1df375c962aa950c766d975aed9a1a2e30ec0bed54ab01d6ef02643a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d5e3881d0858318686d50883a5f5842ccb67c2fe10f7e08d4346b610c82a36ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c53fd761febc27e3fe8b89c03d751cf81de0e77c6754b84534a5a0e5d9d55700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "02877b3bb9fa37c9e72d70f3457b87c05217608f5d8bd5979fc40a67bb39e4c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "792ab17cb7e5e78ca45ca438307270bf02a0ab5af1f27d5e11ac322420f6017c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8b0d88d5348b41171c6d1788c36d6f957a554bb55386ed0372c021202922ffca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ede44b5763f2d9f54111442467481fb8f176479172b795c9983d65a671b1875c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "439ccbf69e6b165aab4fb4ae875537d62151c5c0a6e2015a4de440ac78bb947d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ecfb2dd3b491a9c7f50dc8a89f5b45c0bbf263aff071c8cbadd6d8c89f389301", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9f08e9540a834e43cbf4993810c58eedcb81023dabb54cf6a590d1efee423cea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8abb6f307609d87575d9445707394400baeb699bc2f23cc650e53f97dbc6a03d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4212beb8e2982148f6bd3626c8fb3f6864ecc65dd0ef6a127e8a6a82a90d5927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8aa9bdb9e0bab950d958e2cb37f76836f73cbea0148078346f121e0d87ba8539", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "95fc6f7b7412211594e89a51830b4ade4c9c96fd1bf7aab10677d2e335110fd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4391e90a9e56af280558f47f76c9c3fa3d52744a41d82f302939e32881f2e324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "06ad251cec3d11a46e05bf37ec4aad80fbc61863935d15267b00da909d162ba6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2fae79fb936d7c75bc489afa7d380cc9e4159bff4359dce0e3de6993bef43c04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ad0ce9c1341184ebef171df31bd25ff1e927d26cf0bba3430063a7163f4f4223", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "595e67fa0e0c3d839b08729e984ff9a4703b7f0f1e86bbb4912ab42d576dc841", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d9a7250f95ecdc1ef127e261c59d6ba7994487fb431c807626c8dc8e703eeb95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5e18b546f2a68ec88318b2ad6b161805409f5f358c8e8f6d7c77d7f50741bd05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f23ea3b80fb04f04eabdd001dea88b94c932064b1eb9d76263ddc9c7fabc5c69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a1707afaab7834fa5809d518d45c02800e4349fda259e7f604f2b80f643d1733", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "139cccfa26466655f9dc82d7a4a339e932efecb684e6ff33884d7f91ada56b5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c15c993aabe79b5f666b69da673ba416d1b50555baddb5cc77d0f9c675944ccf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f42993d09b7b473e52288fe3d16b1858c14e7ff57c6c4bb8ee4eb8a07389192b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9550da37058080d54a25e2c573d75548e35f458b1eb3fd81e548f4debb8e0ec8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5a39c0ccb1314d7e96e65bb36845d0393be3dfb29f8619211478628b76eb1e67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "097db8bd8af41a8cd93f820fb0213a98e45c8063c83c27c145871b6ee16e92c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "76350a2a2633a7ea1d079c18287bc16f7f830ee2a6b5710f8a58081fda450be6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8e9d236b2e0b7fdf366c8b8640cd2df844f3b2917c7ebc0672e1b907659536ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1ea924f9d4ea3b91185472d03d59a0b4a57328c9341469050ee4451fd56ef400", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "300014bca1d766f7c369a9ef498a0b7657a35ca5e57e0040dda7fca0c5609f80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9298ee19ff6026823cb4a9057edd5ca84e882d0b3ba57465d5c1def9881f88eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2033c1d1d4e6a63b22d24f27d8e6bbcd83fba69729ef9160de7a3a77b0603fac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "58eb8092b546fcd6e27213d88b7ecbeb7fe1cf0c85c221591b0b75c9360e650f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5f1686898c4ca47cb1aff704a0964ee031afd2a383746a1d5252ccfb2f8864d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "deb6807ef4922e84effab11ada55af8ed021a13dce848427014f00e309c5e450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b2c4dbea2622a7ee26cc819f5b4256bb117db1bd8791e9a3066fd8bf7a7260cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "210a4063223921e1b012e7cf778f9f98780605abb03dcddd63ee877e9ad30355", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b80b5673f3f39892e6a792db97f638f007207f7a9012022af66b2a07bc832351", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5699c42eb0fad8eb73cd8786472c50ad40ee248cb99fd9b4dc31b35d3234cc5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7c4c9f3e57ea1e4c9f73019a4ed9ae0f7e480918b1429453c9a7e2481f39a2b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "17472fec7b91f20ca2bb29dc77fe801d4bca5f3dd086137cf2ad7860d680dae7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bf2353fb61542d9bf11bcd907109db5bea0de31d115d54c3af2efc05f6b0358c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1b029c6153c73a300eaa178ae9d3109394f2ee9f08da0963e6052922f5fcd51a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c828f0acab829b6d79cf9468b98774adb4490a2ecf64d06ff00f9b79b0ea8b14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "17476c78fdc83579133ba96bc2436b869d8ebf1b274825db5001a18a593d51a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "70eb4f80c1910482d20c00448af4a572677ebda316bd17a9c1a2ecddfa8626f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4c0880374901cc4fa55a607f636427165a8d402ab958e059456e7593be6df5fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0520fa3756b07b606cc711026e07ba5fc4d28133f149d5b8237063de1e157a7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4e0b57b8d026e0bb4517a7819d79fc8adf775cebb6eb98645edf04e710eea9bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2188fe49b000dc47dfd661405bb12f91319ca81b190936accf09f8494afd2a9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7dfcb566717c2947bb5562f318212fd128b29544facca2a102f9691a40935eaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fbce3d425c24ac9ecf6ccb796637181cc150022b548645d676afd814919fb59f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c225a6099be905724234bea304ab74dd8e51298a27e15033c19d68d98295f4a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "92bf35d58b0734cd19714dd07d42f1d4d6bada73c5203cecb11ebbe504a4b939", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "620d65bbd368d5097872c5b090cdda6e02201b434a5f22922d218c796d658598", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3389819d5ea314ae4b66231395810b126c5f0d8cffdcca2df9ccf9041a656a40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "26921ce1d263e03a4575f6ae6507ad1d7c5f2c4ecf97016cd0774cb6f3bba86e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6026b5f2d5b5765e84c5597f1e51254ad802e61d8b0bf1f9e57e03c33915c47d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "35453948bb637953e00d293416d3653c2e9fb0062ee78d28e989cb2897074208", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7cdd18c2688d6e477eeda220f6d401517228d2c9b714bed2be23c5d19922d786", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "11acca2d509bb15a9819308a934f7f0372b087eaa4c40b15f30f22261f97faf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8ea00641319b6123f93279254c9ca6e650cc0c3394cb6bb5b7c80a9f0e4bc299", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6503f41f25c6b907995507e42713d6bc20b15e6101c8a4044fcc74a8f12988cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dbb169c9b8b4367194d9aaf382cd6813d5a3b4b22db75ec0d16ca76a0e8c4adb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7e5fe68aac338c8705ec2f95f7134bed893f651ca1d6a7945a12ea04201c36e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "275f26445d370a9ccd681edda00dc3ab1d95cd30910a9e539fe45b74a7f18640", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e186aadc8e7fbd0a93977589ca2c86825f185cd80b59e45b796a729dc831db07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7ce4b5739c2cca5e511c8c24ea1f85f002f1feea76dd00935b0aa59b230e96e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e887f24394b31ee9aef8ca2fb533f4d52dec5f792f5337579e5e7a22f01b8dbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3bbb2c55d60011cecfa736df5de39e018098013a3120fea22daa7c2fb4a3bf45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "af1035608de1d0c5048cf5c237af159824d619529989789682dd31708bfbe2ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a1677e9a59d0acca1c6ab1424f5fd2b8b89808d0539ba9b3d319e73e3742dec6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "71009cddf37da797feba8f742b67dc75d55c36b8ada5746a0f5491e8a6a9114c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "720780aa76a5e9c95df5b88d61be2894fdb27b4929c6228dbb150ecd4fd90db3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b4a759354deed68fdb1a033ba4473ec5bcdca6f45c2991d14becb56dca075657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "391293c3dfd204da4e1c813410ecde39420918104f0b855f967e97c8fa0848e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "067c1c7a5093be4feead35655990a4f1b7eab9c8a17aea430c0777d8012eec88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "72919ee0890468e29febdc72c49d815054c3f843c39324b8a43450f7ba3c4a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "54e193b9630096b81528836f2d1dc31531fc90d8568cf0b2574067ccbc0b7c06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "48795d2498d552571b1e6b036158cdbacdc1e168b50736dd4db97f80b852c1a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0bfc3d4937f3f94a723a774a807d8b771d20f22be4ed3c8d2cae440d5529df70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "27dd1b1a9b45c1925ad7dbf4ebcfbb946b113a6ef2c5635381bebd97d50b006e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f94774b73a72e37fd2a9ecbc9dae8c8b079dbb399e6494e17ac3b9cfc49decff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c9bd596aa1a62c7ab7701d06b602619a423fbbae7111c864b4762187c7a833ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "200dd0cfdb68d35a239afd022c51d5542c3a7175804ae9d6244ee5cd23037cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8955739ffd6e0c6d6b2935141b2d1f4b4f7be1e584df0ac366763bf18ba9d7e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dc669f2bc2db6e13bd8cebe5b86d8e543ef4d3ae1cdcafe426f926cfbdb59b87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d687610ad248dd2120fb1c74dae771f1c7c7cf8924346ecf11707af82e87d15b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7a40791c942401926280276fa2c419c59b5c546d73dbce489c5aed43a8809904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6a7f63ded419f941fcce87a467ea5e604615930428553435781fff82ee0a0987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bd54e1e724681b564503b4cd104c82534c6cb39d3d97e8e2a7ee8f861c406f58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "44ca2330b53a44a0d9a701ccd126c46efb40ddbcdfca1d50bd9899b443f07a8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b4bedc79367114b5b4ef49827ca01ea6eaa6de3829bb3129a98a745c96eb7d7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c09cca5289407a1208f2ea497a1f0ef42a41cfac4ed62ddd6912bdbd9c310e04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "104d78f4e1e1a414d8be372a534f799b794420680ea55f94a95986868c09464a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a6e10c5b20e57f32428909dd2e6c1cb54fa392b6ecab2579feb7fd317e321bfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cfb1e57918831b1b202b22dffe6b54c29ba5e9dbb78e153aae4bb441ff475846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c159eac67a59d611477f71d4587040ed7c8bee134045339cb6edeca085541b01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "374eeaf107f0a8689bfe3f9abf3c317fbe4c12acdbd7276b76587cc0581ccbb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1039f2e494dfa2803567df0b0278e4993cb42cf6052b073d8c5cd2b0c508b053", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2dffe3ee63c562b2275466c404a54322c9f6f5b280fc8b81d12305fedaadcbcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c33ff6dc347e562f4df833cfaf1cec30fb8262b413aa78388f57af12213639f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7a86f272cd4901ed20faf0b429ce72dcd1933bf1a1cbcf6fabcccd7807320f68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e8a157225d9dca969047180b738c0aead5e5bbc433e17b0bd7f73ddc869f4950", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dbdba3c63fbf9b3855076a5729c1625a99aa2a4153c8e0e1d071f495f00a0f0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7ebafc6e6b51fbbb4dbda883d966a30bcf6e51ea198d4251f9d159b35b9a1c7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "36863ff765f095bbb28dec2074bd36e280ac94b8250f7607e37b3e6bbb88c220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e9278864b44d2f10bb1357ce8150bf54307c0328fccf2b40836920bb909e7747", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2e213db8276898749c4661b9dd0f3e7078fa710989a1ded9869d59e73e2f485b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a9e38825a689a85f946dda8eafd4c9a10c143b8c8acce0fcfcff138d057ddc9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1bd2eeb19ec68e2192a736db6f7091cf8fb4c6b0130fcc6018dab0a558327158", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c9b8a405645639e3849165f4ebfe0655b138bed58a4ced8ba66509427c56c161", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e72c8640caa43110a70829ae330b0ed37d9392fd070033d7d13c96e381708017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8c77497b1fd7bcebcad989911c868f508e9ff98c6381029205f8c697d0d49731", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "de9f64f7a48617d7904caca945b98591c486b3809c17db8fad8d62abea4c1516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cfefcc263fbe15828d6665518b62b563bcf9f3324e105cb36fa37128843c7973", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6705f9f9c3cef73ed3f06d20b90781cbfe00c344fde87e474c4fda315ebffef8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7882cf65cc8d15c642443edf3000595040a76f95d01bf7e332489b6e7698e89f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f9abef22d039ea2e9929f9ffc6df6f625a90e8b63182394322880a6b78079f61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "27f311cc7aef44a0c529415afb788c7fea1d68c97a04456104a7e50c28a4248a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "172b65d4b71904a78004620bd0cb9a0b524c9c4ff0b0735dad59dabd74cbe32c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0cd445273b40afaa6adcea36d52a67920948c2143e3d59361f1e49b90b7e190c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "393a0af800df3e3ab45e99e3dda8163c071d9087c30b75d44b89da82f1efce3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d1ad936ef4e126e310c06067c660374f96e523f836b3c40bcae05ee6458bb1af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1ad20ac9d1e26b1f7f71235d6d93b536611810bfc355823a6d0e11251b3e8ca2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "171d5bdb0538b9eb0d67ba12286c87c8458058e2f1ff70d3494256ee1ddb2f32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a89f98b8b873905e83a7bb008bcefe12b1a31847b62bd381f6762c80c3f781ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ce7e2ac234ff58adbdc24b5fb841b7bdbf8d928ec4e4c4b5e43b8e9e36d45f91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d665c1373c17491f69a51ddc9bf4c8488b8aa51047a943e550fe1bc1c83702f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4a051f7a5d28db63a79377709e450a17c3c80f14395072774eef1a67a3a1f01b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e1178002d995ecd1529c70989d5e921f130f90890723c6dd31c5b88015588baa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0987e0290af7cdc74bd094dba3c4e9a117453828ae90b8b8a5edbe268e781e9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a65480f0d05d4e09fddfc959b600cc5fe6e65394d806fb145c1cf3c3da45e5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "70e2b3d4980c652ed4e723f9396b9e70e980df3e932993242526b7d47f644dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "eac857dfd369b401bf9be3e05f25c1341deabf3527c5addaee4a2ecf0f34c482", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "afe84da8d73760f9082c7e297b721cf87547e5f7013d8a52308542843f49b9c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0cb0aa7fbae10e10a952d1b6c45f78710a5c93c0bb29d05fb0321e524532b82c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8cbf634a6fc0ddf004822d3ea70c1aff873b10686169c7c7d94c4a85234174c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "65f59932ed5e191ce68e0df9b4c5f34154ed09c52ddd9d31b66b80178d9cccd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "61704d4084ecda7173197d3abd1e4e457c65273bc16cdc0ffbea2a62a14c31bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "daa0b41321e1c468359a18b6a804f2b364c5b5bbd64d762a4ddb12f6cfe4d2e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2038617a8d155caa3ccefb7e3ecf0767b9779ce0d13d27f1365312ddf8f82c08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "779c770249e9bbf387a9c095daa601280b19fd9b053ab2a9cbc40dc37e2f399d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dacaef399b70c2b2bcb60db2c088ec33fe5c3b04c390ca075d223d2218ece993", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9351fd93abc523e646f2dcb934552c9c19be6b6adcfc1ec25eecbfb05a42a547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fbe363bcf979ae44e3257d6aa3391c1dbb4d6308dbf5b558d142e1babf690295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "394ab3b7c6c026135a4deda837f5571645e80b890f93c174bfb31115261bec90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2697e76f62e62f7e6712a359427cdd81fbeff269cb322371f98906f968c60304", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ed030df5fce03a52273eb58c17fb8b42a2504353dc252847ee1c98cc574cd7a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "195174fc5be7d9c23ac9802fb45fa5a9f0e4f3e3273dd3673f7f499d6c4d6ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "aa1ca3ffd0fb9a40624e0567444b2f7cf430a81bd0923a5237284ea7c1287b80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d2adda0a44f092220a086a3371b129829ede070d268d4bff01cb81920c2f0209", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "43ae41cc119479c5df6aa363b9ced48e570d0d9c2951d28aaea891becbeb94c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e8bcd866c02f1dc2a060fc5bf0e5a6e30e78ccd0fd29d2d1054303b1963e3d22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "74a330b6c327e3260721e7f6a93bb1aa991bb9374d9107e7a67127a66d0cfb8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f34ce1c9042002654600dfa3dc1053dd1e5ac387e307aa4cf2bda61abfed90c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f2932c21dd42d2edfd6d3290c52e5e437e0e6c46e8edff8ee5b082d282faa7f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fd13655319e238667e8c801522fa6b2a7fb4cab79ee60a670517b206977f2bda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "2ad86ad48085dac184b66adf4f4a3984d02e10d6e3fed6b584992439a03ce5c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "04cd9a62cfe9165c9beba59be4a1afe53265f7ac923e97e6bf92f1865c35fd53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "339bd52bfdec7db37b27d050319a5c6665a0c495b2bc2b9c08a8edf2f816f65e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a38ddae2328c329c58b3050f32808d3207b52425cc07adbd7b358573350f6dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8111a6d74bfbf741a30b0445a8804d0d25cb63d2d8028aff6519b8255dca3905", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ddeb03a4b5c5857c5e57680006a0bbec3d746b2bdbdeef1867663336468a011e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "75fc43d47a79cad3828ac25cb20abb52e2c1e33d37349b4cca4f975548bb4937", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "9aceed89001e75e3f375ac2b8e85c13adb7ecdd96226ad1619a6d355a4b15b07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cbd2658c8916a08274635086afbb44231d15826ac9501fa8e4f1b8e26204401e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "099c3613048f801d32c39a32cc43a098e14309ebcb97a1c0d55567a450daa77d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6b6a618d354c301538fd066ebfd6bd33710f51f4c751653f177a28c8ff7875ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c699fb1465f228fca31066784eeaea1e2271fa849c9f062aefbe986a9c7b24b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "85d995ddd85b1893efd2435866520906b8291f46aaddb9eb5cd879656a212a09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4298561cc916637d758680834e08e88dba93e29daa43419e1b9d8ce3bf97e03d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "242b083ea9b17d15fc156e7900551daec0c43fb2b82b97049cf97082f476a26e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "806a4f23bbe43e8d1169da28bab440191a1fb8d6c86cdd8806d21de97fe97898", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1bbe948178c6535555d8588315a30f75f2c0140db9d1510f37cf5531b04a9d52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "20b471fc3d1130465a1eadab24efbb0be4d120a07416752fc81824cd3140004a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a5d243b6d171c33997de82f2e5dd4b1d9a9173f02e2ee3522b4b5c09b1bf226c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "992487ba21002013d73db0508f0b3baf1983cb070c88cf17f4520197412bfcf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "938dbfe7c7c498131622dfa00c95415009f86de2928c29a883ea9400b9b35933", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "009d57234623d2943dd6c0cdb6a1f68daa955228ca1fea47af269fee04c20062", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "80ffcc20f8b66b769666d2fee5233ffed742404d37c4161c671f170619518fbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bf777c7fe652cfac46f4af26cb0af88815205672bc1755603ec327338f3ce440", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "022dfa27d353f5f50091498c69bf20bc0c60e03a81b3c46dd9cff382ee5c19ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02b8b6efc10e0b7e4a43b61edcfbae0f2b822264689848cac76984da3daedbe7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8300fdc9a34d406605be31f8699df1193e76fb3de3d699c9d2c9e5a14c3e3bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "66125c9d857aff346d66ec7af6544b558fd860eaadf2ab04cf7ceb45fbac0255", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "15a67bf07526daf5c4ebc653f63f6e60103a367cfab18a717343eb7616c4cb7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "66a739683a5482073ddcb4fb90979cb87baaecdf9d6a834aca1a143a3aa2338f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "823d594dd2f9de021779135e4e3469a250b8b342e36df699e4e0027083363c9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "41ec0762c8499ca187b777e127f7c848508f780053ecabfef127dcb0b37ec2c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e01973f9f473574c87be880be9a459c9d3df49d592b7a09f02351522c3da5a70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bbced6f0bc7c6634a5a1bbcd0d527c0a2468174be6c7f9dbb168b5a555a6fa6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bd2ffa172c0c096d8f7ccf7118dd4d88ec0dd712b9ca2839cb009555f60dddaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "13cd5867816480ca69e7cd63975ddab94cd65ca20888683624ebdb0670f75fcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "817564662f38c578cb724976989726a58857a5d42c7eeb2e74d5c31f919d072f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dbc4688d6c9133b11b2a56a1b77e463cb07073a6d7077a848020e7c378eddd28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "21af62a33b18c628bbba8fe011b9f617d6068603e36e8455e772382849525705", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "09cbe778de7ab520ba050eeb6a498534a99601917b55cfd5f89e050fe8893611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "185694e301a7a71450f8a3d3b46bf06edee4a6fde39375d2cce2a730e2b3b3d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d078de798f712056b994fe954ca9fcdb8ebdc056f7d8d0fa427eb2fb0888af0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c430f0ba7b81c24ec27a4276eda32e8c3889b3bdce5a822b3a04f2ecb546ef5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "73badddada7534ce0818b25814a2a19aab87655b98dc9113d031ac9bc71e1fda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5ad6d9852a90659c1c0164404b25565ea12546a68a4e9f0243c841eecceb3e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "41f52a87e81ccf88391f1a242bb36aeedad54e7fdf07a737c7d22b65b6fb366d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "51621bdf7aba90a91ffa97b346974df6550375f8581b5f4f820d60b2f2605334", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7414ff122502ef6f8d02788c95aa57444964ed3fbde4712419ba592a812b9fcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "659452c5eacaa0caddc62dc90f75fd6b434cbba0db9efabab737c7ad6c99dac8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a3dd797d79a615169188b1cd3322d0e8bd6a9af2f547601276d0043ec236472e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9ae70772de209fad267244f382ad119004f55ce822a23a905cc34b848c47c030", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf90f92648ca9c5e90bb5114a1de3504e278dd572eea00edf669a62279541178", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "95742d28a6c76c3de867d181595aa4c13f3b4b832c5812c21265298be923cbcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "14a2292d856019ac081f9fc7187bc1465f4a47e5d5c73028fa735a4e1ac74c02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5ec287877ecd1957ebaad8e1e7fd3d6164fc7e6fb6492ff11768bc13d1709763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8c3d6f0c78d1a105d5c153d937648ad7da94b0166a78fc90619149e0d5ef01b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c7489687e30f213cfe799ffb2bea576ff74ea90d0aea6e28aa7831628100d857", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4ceb355d97fe54c99f04b8a7356211878aeaa9711997cd2466ad09f42904dff5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "267a4253a1970e89cb9c1a6dc9631471e1f8f87beef10bb1270e460d5a693d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "266425ef9c3cdda5de4ecc1952341ed626f8c0b8af3ae381b23e4c54d8a581d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b370d3564406cbb1a00564c77360e6bb8dae48f89d921fd945aa013884932002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "32083646cb63c3f852d6f3d9236ab68bba5d2839f97024ea824d0a3f4a85930f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f7f3a53e563605df22faaf906d4a814c53045cc22b03193a1d7ce6b9b9c61bb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4bc86c55d873df6dd5a57db132ff07e17ea853c2e8d20ab0025ebc64bf177aaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "5d575502464c9ee59490cd8e44f6fa4fe293d437df9cc7a1de00eb90e9c6d66b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a51b0fb9ebd8facaf5e583af03fd7a4accd52af62ca8e4581059192c975f112d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a112de6c3625e4403418a0440822fef93ad4bfa1c5a4f1755d8933c7f16d3c80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "41a245b6439638ed05ed7944d9300f72e27eafc854b6c163e80136e062a4a9a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b882dc0fe06f24a27d1fd7fa4186fb5941b4538396734b8a53779e28870932c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "6f4eac145f6c12b095b5e6883045e13d25da97888cb39fbe921f2b148a515dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "776c85686ac6174629229be6efb3406ab77b607e1f41a11e16ac19ea12a8c13a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "55cd4fbebf8d8823dd3f8f773760ba32a271b4d790b3f3a92c781d5b3fb19d0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "22db34c8a77afa5ca4cd7ab70a3662642c3f8cd4fa2c980b87bf7795d9b1a1cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "2825ba17025e1d39cf64770d82cb48b5233b1b571d7cd102b3358077b9122e85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c606bff4eb9513b16b603b72d654f081cf82c01e9c38c8c67cc2118b24a6f9a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "84923659d807472503237345de47f34fdb92f0f2bd6919944c999f0085f8fad8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a61635acb3c4c327695970eb025904c3b0956515b5c6d8d516fb4427be7215b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "0ff75694687b594ef6cf7dd77512b5496ad233047acb204483ff635d35641778", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a932a1b5a441e81b9faa29c6a6847d87d96f0cfc5d17890c3f19c7d1dd841162", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "74150430a260ba994e2a3d3d7c6f2863bd9353f2cc6893533c1f4340fd54208a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ab815a68378ac3882df40f705caed807f5090a52a7df8ec3b04f428a7c63276c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8086a29020968292d98929e5c4a29af82512a549237d62e7856840813c89a28e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3584b0b67fece5d737894e468720503310c1b9ac569b61657a86461c056c73ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e7df84a3201acd8a23e0fb1d21dbcafc4aaa65937227af4b66d9368330d5b597", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0d4cb9c5d781a2e540aa22a3d6660206b60abb96f2dd7a8a0884cec8e42bafe4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "29899358ed9bd215f8fe1761c73819d163eaceab64dcee3c356aa5b0566f59f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "735499ad331e97bc967744e54a10280422ceb683af434d23092951930aaf9f71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "13640d60b31e6f2b6426285fd74803b1f40c55b662596fdefda00e1848ef5f79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "06a4f9e5470f804b9880f613ae80771b75a7fa9ce4bd9f286a9b81aaf0a4a336", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "b92539bfc452a733357c132a1ba4e58577dfcabce31d30b15cdef0de6295c8e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "00e803b7793c0977f954a993e86e1a5213f538eccbe9fe33fd015334bb0f9c0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "0a42f612d73f2d77e37a86fae5298d9a7793bb197196b5ca36ebc67b0082ae27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "af2b6e2d65304be7ea8a90079b4ec521f68198da2d130ecca1ad2f912b1da91c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "13923739fd661cd2d32bb0b0a94561232ca6f837bddbf0207efaadbd9bc2e99a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "b6e0ff9c47b1bbb89e037f102b2415bf1f69e6e892f0faa14555d934829219c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "df6a259aaa869c4b5b0467559389dad2dc45560e77a1557d8eff1e46c78546f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b0bae027e66e5125197351162ac10dead6adc46bac31594ba7b510e057709c4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d0f3881751f335c62e204633ed60a8d1944937a959e270537b09ac10aa028c8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "83bd45b2064184de8369aec6776eb8a2358e5f2c00205c5de96463c9f7bedcd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "5a3108c9257575b50a39392bf0a232f6bf1b3d42e117b7b1af7eaceaab48ed60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "a61d2621856beffdf991e0c7707cf9ee0fdfa7f9dcd6b1c6fbab90f534014b3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "be306bee05b796f26a5b2acfccc832ad25bc07371b0ad08d2d5a402ced7aa48f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "4dec92c7dc7776c6a8b04b9a3340ce203c7124d003e5e6e4cb92ce759570064e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "465feb4179c7b89573b78a294b671d52db33a66f352066e9b4086c622911b8bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d6dd02eb19e1334761547fdc8f060eece74531b0e5635113e1b82f47a1598f6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "5ee51a99546d43c82f0b2b0484b9f989ca5891b8f62a9791c5cd86205b981957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3d0cecd394b8b9a6d086b6bbeb520719a83dbe881692f6d41a1cd810d4a42042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "44fa85e5339e4e65412044496effe16316a8300561a29e1b337750fab95ebe40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "44cd52c6af1d4a7ee39828c2d6f92974912ade6b2c1d9ef8d994d03a78e60f12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "435f33d9b17db78817d5a1a2175aff4c0e141e7c38f12100a12d3fc5dc168281", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "0c084b82d99eb70042490ea7d5a78ffff0c68186f17c980fab60d82dbf3cac8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "92c37fda6cad724d89e365f71f2afda21583b52a3ee7d8117da8944fed1b13c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "7c06bccad608cc368b02983b11517c4d4499af2b1477a274313b6c9a2a69efa6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "ed4c9f948c7edd1ffb99e209498d260031c085873ee6201fc9f5c6e07d1e2dfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9a43f4ae60ff416a4fd09f11f2a6e7414147e4fe056e70724377e370b472c746", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "190597653b58109b85447c9b19c2450d272142f06173dc936dca63a6282e3712", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "e9e1b204e8fa70f5ca664d17bf50ed85afde0160f4963cbc9a093ce6b7d08515", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0aea1e4e4e9b8229e934974fc6dfa6393728e42cef19a1a79ae10bd08ae297ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "864ede9f00f92090e9cde4cc50cc27ab244ccf82b9e7d9d827a35d39c5cfcd87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "db7bd6fa00ee2f922718327048a18a939e35907bf1c043eac3651bb9e4e335fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0f5392bebb10833a77e69b60d1ce0a1ca99b3c6a2f00cfc69864a947690e0e14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a7f9ee5783f204185992e191355c4f6870cc8b5cde1d4cb0229a2fe1ef7c43e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "83890bb5de472efce624889e295f018410a6d15b5e38a4b45217a59293a278b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "01880e828423ad8b011bfcd3120e686e75af419c87413979c56e8e7f0fdb1798", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "8bac2602ec242043806863ddc05bc3942f3036c14a002a50ff94dd97d4cf1d43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "ffbe5bf99b6e4e36758b915ef123cffd85f82f9d397f31da12b932c501806e72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a51acb37368e469c768271c8cc0580b5613711c1e1cfb883be0972deafaee098", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "10498acea91732a799b883233c58db1dd855bbc2d5018e644bb34d42719483e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "0fe2fe3cf62cd95943fd5043da2bd9781f8f4f59dc4e655e29c0c6deb6fbc0fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "78fe01b47d35c092e197d73e1f40e09b1cb0339ea49c13fb79887ee0c6579b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d068867aa1b959613d455634e6b123ab21bda4a3b4e52f21db6f0dc9d76a237d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "e036e0e75cb75eb08c48bc10c4859eb69821f52bffb285c2da61a8a527dfe7e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "3aeeb7edecea80f0c503eee5752c66b878a13e17a17d30c9e6d1ea945cd6c279", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "2fa0536dbb3e62995b4252f236abf8dda094e4a8fe1a2b46068a55feae6b60a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "384be24321fc99c9eaacd848954faa59ec385be6aa5f7f5371e446ec738ff025", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a2094b5a452e1ac8fb54092f9795f1bcefc9c16563ad6b9b4eab968042c4c5e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6b04b3dccde5ba00d906a39bb04c57a1be45e6447757ded6c95c239df0e9c0af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "0a3850410bca02c5980578b3c5ce292d193645092ac5ac3cf98abdb9501e58f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "bf9add12adb3a2ef2be560cce1f0875e2f2468b9a81aaa888e5012d8579fd1d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "c637e3ef120b9a6bd5068afd3f60ecc75d01bba8af4122cdc301c33731f745b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "e8e3ea8405eb884d2c247c5c8b63327eba91422787aa355d6ef79dc348b05756", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "b1ce6c242922849f65707c4eb8a4d7700d38cf0f90b10674d2feb674a8163b22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "451ae8fa75d6c9ed06bb009ae1c72ed840a86d2b3f035a5eea49e697ef1937d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "c50c2c83b19ef70347270d96f5976e095471287b13c0abd9e6732b6a2b3e25c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "2ba5426b75a62ba4d1affb72686267b465d8a9725bf2c5f4d8569fa6ae0c74d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "ed3de4c557a5d8b2918d6e6f662cb289acf2344d5f0629f1846604425537f349", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "7bca0bf96ec6fd1a8ca4283e719f4073278f1a799297e467fc3f60e4919eeb76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "02c3b9ac6ccba17fe16059b6049c5be1c5b66d1659d7759de9066f57bee2fefa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "951c5ffd42faacebff2aa0334131c25974a91b4526e28163ed79d236ea5206d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6b6227fad843e34e117fa8445b4a1f4d6c4269f3e2fee946e2a89c073fee3726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "1743d8bd9e8a6053390e6ecbf26dc196bc5473c459ce7f41de16e7a612ec19a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "49ffee183375ab5cebdb3396981e6005a5f1dc29f8033cd98e8d7117dd000fa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "2bdb8f4aabaafef6cb5e41d0d14479e482386a9c98d454c3f33c5e855ef969d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "fbd500a46908f8dae2047ef6f2e1f78526868cfd96c95c333aa568ae176a0763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a7b731652bc6f1600565e3d2cf59b3a2ebdd2e0341bf6a82415ea17ad2958ca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "2bb66f3fb08440ac76c0955ee36eb18c6d458d572bede55419b24209e2fe0431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "51813bff84467227edeb3580bcd86dfc25add287a2e31c40e11ed1220e83de39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "c211d394e8b14050abd900d890bc973c119ca9f3d3f86c59595b8813334e5c70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "b526fabdb033ec9ccedde8f3dbf62c70ca0283ad08940fbb45637f127010ffc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "b0bfc0251764accefce8e44cc8797fe356152e137827a711ebfdf2837ae4cd36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "1c40afe852e97f9b1af00e2549abdff9985dab0475c4b3acf4a3c3223391effe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "5fb6eabc9d0d95ac9ff30d366c9479704fa3774bb09dfa12c7e6d82a6a096013", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "262550d1304bb43086920b4469f215234c20d2e29c07bfe6bd55408f4ad252c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8a66aae8bac1b4efe00e5742492ca38a35ec4c2209e1a408fb5269cce54ae49d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "16683a266d95aec4b7accf37fcac6ac9aee22436530fb84922f79b8acc7ec9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "248344328541a9b28ff92005893766df2d3eac61208cfb0d8d63db200b9b7ebb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9a19eec20b67190f31b0fc66badf0c70a698184c275c73ee0057078b0d38aa87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "83011c9942e41133320a70ca08812883ef9e98a258f31581c27d7b5603589be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3ed799b2b9251a97d83170f4449f3039e36390cbdf5b8229c4205d45adc6e9ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "bb9c155ed876de86c9234e99975d0a55ccb7f664c6df09750aff02ae9d91fc2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "ecc4c3b565586993f91f2ea2eb649641c2bf2269dea2a72ee5c2badc5bd3f53d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "4a019dece99023b1e98fb431803e0b1e9f0c2e031bdbed483ba7ea334922b719", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "05e1a47c6c7bb7e7b8a0589370af4836caef1067dfa0a123d8e6f13ba39fe18a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6a52942346f371713a9dc3105cff1247f468b73c4d806c98c06c322e37b58a99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "4b57d7c6169144f080660df3c54fe42e038272cd073ca78c89c5c5d5d16eb2d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "4c703188f0943046b97efbd3c1db4318451641435c1976e40029af3ee0d8d14a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "52a0d898158c76d99cbbc84c26c4c729fcbafa360c7d9609186b1bfab287e056", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e1fe5995bc5f0c94fef285c1c77fdf2f879a7e1a67150c52714378f574466447", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "16db97625d9c8de1056dd6eebc21260a7a14a0c728be682d8bc35434a10226fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "43010ad4694643f1db7383b00cedd83da974005e68199067fc2fe9e17bdb653f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6e8d05472328d2cdf5dd2351a5d6eb5d2f7cb255ccbe3ededa90b5e973d1573b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2532cd0d5a2e1a5e56042dfb0df6c7d1d8dc1a9a840d083d339f57343f44e5a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1cd7af24edd720933bc417027c0011ede3e8e55e569c587bccb6c83939fe78dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "1dc92d3f3874b8a0e7aa19c8fdb10494016c97bbadae3c5281e19ce2e1202b25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1f1ef4598d30b1af7eea5e7f297a95fa7d12e73d512b7e4c86e4b1ad497c1ad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "82c94cb2debb2b10f5ac9538143d14bf20eaa315d4007ef94f20bbdfc0d73fc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "2972c2223c62f3f9112e9a95c078d75658236ba478b1e5fbbb871520f04da305", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "928c54a8afd1c3b44df5db0b36ab08431220025222c77482bcedf77a4393423b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5e31abcc1a7ca084ec1bc372a71c86775d942b9b3629be0d8f3b2e728382a87d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e289d73fc44b5536fa7a0fb460a8e8c30bb2ff47447585ca06d3f73395e96459", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "66c29256963cc4e3ff9121c846fa58a84238747dcc68800e21043536b23488cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "477f1c9849ec9dcde802e52fdd13aad9f401bf92d440724f68b55c9c5f0fdc1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "9dfb7d47336d2032b63d016f0805326e9d308b8b5ad249093ef10cf0818edf97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "3c642b5912049ba0f782a25529f55c1e603d7e295d147fef106e3ef42c8b0f92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "88b78cc6dd1c92da5628619865efe235b4e743d1f0838de8e7227918abcc6dbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "30d0c1d2f6927843f234bf1aa9c40b6ec750a3a8ae60ede7a98475bb4c47208c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "b9fa43d4f771c976340859e8702bfab6aa0da5327ae14aa9dcc20425b2b43d5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6ec1d450d1aa6e6fd20530d89e23edbd94f606afb807b33f5c19d387f1b913e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "73de522074bbecf80b207a5c2abf8346c6069905c123b632a5393516c53c8858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "4ab8b692f87c16b3b080468e87a75c280c6b6008d51750fbb71014198044a7b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "d06230eb436831d03b94ae5ff6275faf06d17c66e58720e8f4632926e37aee80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "4e4ef7d1e9575153b10fb38c1ee19f8ec6c1e257fc1f5d7fa80d78b14afc8f69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "1e8a5627f0095348fda5943d34cb5621aeca37d8eb20da69ace18421fb42bfbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "4da3921e1da5f67ed3d4ec973f88cbe3f41c711f15516393a0458f8a01067039", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "d39a382b1894696b922b2511e34c95b65eff2943d7031783f4bfbc4dcfdb944c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "189594e64b1fdab1f5c5146b800ad3c275b9835d77e93a2811566447bb3a91eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "1c5375d7ea33120fd2366bf8bcfc027040d0f27a6dc76d9ee787d0d67967e88d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "8e4eff69b30303394fe3ba7560ff078b4b76e55f0cbfb7cd636eb544e5ba99c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "efe612dca8179e94e3544dadf40fa9d56db51f12a39bedba5824896d9c47a9a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "087dfae7850a1ae052a4be384ffa4135b8dc5e0cc5099a87ba77d609e6924630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "f4bd544d1a5e45b004c69922e20420e5b94b0b93adabddbaebfacdec40229a40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "b351713c0e5749435a3327d0d6332059ee15b7bdb8d0ef9310d7100bfc7ae400", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "019da224fa2ecd7e12ba23eb41a35cc6ff75f279bae380564b51703d91678832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "516c4ffb62072ef96ef12637f60e0501cf123716637700b1cc878c87a452d933", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "4302d29d720f211f208cf83ed2765a377a66b1f5aaa788ae990cbd342e18c028", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "b7c2ca1c8934827cabadb60bea32bf7d055de0c1021fe10657a1d4d142c76a94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "0382e1e3d342d0163d7ecc3d647b30f6af9666b5282b1745c3dc7fec51db2ce0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "4dde1cf32e3015da3655b53c3377eedc57358fc7566280595ef4c1ad5640ec3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "3c8d77e2e762cc45ccebc57f41289db1573ca9f040268628eb9d097846b508be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "661b5f89813d82ea5083ded97baded5197853b01715a2e401c1e26aec44dcb9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "9b9aba80eab856e041881c44ec4a85aeb6db90da450dd8bf607f707a670d3ce0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "eaafd9bcb0ddaa89c39698708b83692e0e6473c73bcee2f049103e4f4d05ab6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "8b08dc962485ba21009464c3a061069489fd12d9c2b0e9cef6599930395b9c50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "7ba3b7e161f0bbd2b50d5f182e3da65ee9f1b7f4fb7e9e252edc1bcb5cf6d3f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "774c637063dd97eb553997fafe5ccf9dfd318a803b7fcf5956dfa3d55034be37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b698fc3689152f1d6b190a99a5f023d4227bc0b4a7555873b79e9745ccafae45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9ff077a9f7db583b943c9344962357dd4d0dc1b2d612a553376e36c29b5f2e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "63fe21c1510b429a6214eb5a1cf4341949c93491fb2fd61382cb26e0b1ace5f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8e36ee47a1f847fd521f308ac5a5d6db7c609d11d53182a64157db202946138a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7eee4e2c2f3e602f9cdeaba2b48db33d101f8ea6cf69d2400bb788e17f82b767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "d34abdc47556432fb2621ca7b8269b7d39b3596b6abcac006c591229a222f894", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "03abfba058cc048f42ad6b961c4e42e74fee53950cf32556824699f53ab575a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "8752ee73974a1e7823f695a1a2a9418fe093b5d3fc0bdde149e646d38c60fc39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "8d2f52a98a6d28614671f1570eba5efe1bf2ff26ac87459c14cf56de43db9d87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8d9b69b6e0e27e6770dc7ed97b071fdf10484faaaaa72d015b931d5e3f19e588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "bf26f2b1295c9c3bb98babdeb311e8cd6797081a1e303176b51d9866510de26e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "bfd944b2407f71e02d14ea31066350e3d4f6b80b17ed806586913472b47edd25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7b360707a3a7ccdbb9d64cdaefb7957646402cc75a7bd2ce5440b325d06ede96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "afb1d3790576581a535db354417d1b5df9793746eb158f02acfb3034969032ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "69e9d5c90c09b70f01f242251e141c4df60756b9ec6e5ff8a7b38d94597deb08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "fba81edc7b0a880e5e6a1fe3ebb0169499220ac8215b6a35ac9dcd6fe24f5388", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "caab126eef04da401e8abe9bd1b34ff31a204eac468e5f55b2d296e6f89c6a29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3a87823a5e466a40382bd865922850255bf0890ca6a091612932c358c3520e67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b86266eac909b5a768d901279dc8ad99e8f29b5d295402b61dda19c4c27d7942", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "9f2f83008843f2f025b200460601554ab01f9e3a963caf46df961004a46819ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "cb8ca219953b3ef52322a5f4f742e3230abc17e8d6f46aba2c3981760e5ac858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7e3bc05e2952d3abc1b0ba11e08964f15409df7e8c91b175b2b132d3d5384d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d760c6506fcd3bcda71c853cb513c088453d7a8d237c94c239614b18014520c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "0a631e0a6c008d816ed28f1e02cc6c102caae1844c229ede4d3da05abe416ea5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "588eb15c273639ffb5ffe04dec57772a0c2a225322070f4ba30bd9c959e7654a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "8836098b6982f6eac4a8ffb545d96409006a434cc3f2e459e1a51d3cf477e8c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "f6f240f3b87cba66d7e070403077a0f90ac762e7565534c6216f8dd405bdfb42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "0a4a551474afd3efee30d0f3691f4fa16a251c64e25be5bb15a9f93983c09516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "71c9f731a4e1b4e38542dd2b547d7869c8ae3497920c94f6e9b42e5ae1a6972b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c6f0a327514c3f6aed0dd4025e571782d9e6d1ff0f2a4259a5c2e605d1394cf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "d66e89a27dc215f05c225d5b8bb6396d494900262d8593bffa4907be099a070c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "f7ab0542ed142bdbd7d55dd23362f4758cca91c231bdff66f1f8c6265a718845", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "91723ec462b1d88016de1dd39067058850c2862901260edbcabcdfb0b61a843f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "4499f02cbfb549c025b2aa8262d6854a9f2900d823518b6f968fb4552223861e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "3912de08d7cdc1ea574ae969be6064ca196b3558abe4f07bf02eab84612c956a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "5d8914db689833a5b2fb54a75ef521d148f13cb27844b241b6bbb79113509762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7dc9414238d4bb1162034abc8d969295ca679c520eda836ade66c2e0dca6ed4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "cb37c77d74643864056ebbd83b69aab65e685bf249b0f7916cc077ea4a643c9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7d63911876e455a6da9d1b8c87d677945936c8b7b00ec88863247860df2c5db7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "03164c38cf35596cdd052f0a3651a1704bb7e0e0845356de7dfe6f1aa089ffd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e7621a12840bf1887a0afb42bacfd5c6df9a5221267dcfa6aba58c4be78211a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "eba32e26c4fbde5c19ccce14fd6d406fa0ee6b365496172d697d1bf546982295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "78bb017da78aeecc59dc08a3a76fcf46ef8052a0c111add1511e8bb57e4920ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7fad36b5bfd0a0854f77f047f379a452db75388e897a85af8bf4ec0a403fb106", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "af03a42b7ba7ac326002743e2aaa8592d2f0f977610e95323ae1c9c56cf0dfff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "b377a1d3c9dfb36e6980b2b999b091ff19a0f88074155f14b33c921a924cd93d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "05f7b0cb309bb4677bfa356d9e19f9bf077dda7bc63df8304e5eecc29a4c4128", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "37afb8b999bb0344837faebcecec1bcc94bd6090d57a52868c3fa5a1bf9a5194", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4af328a36d1949c906aa5767c3098384cc8f677edab960f18352bdbe5bee3f5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "bac581e5d321bed2c2618a949f41078099566b00fc21e5e0c82ac5235767263b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a00f1890750be87ff530e899b9a552e0a9410d2c3dfb4161b77c6228cda43b32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "6c21750a0ce556f9e3ce6806533cf696867f5d2662ea7b9ae5717a2bd884142a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "60afa39a2885b9711387833d42267522e87472b341df9c4ed3481d64514695b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* 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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "38b5fd9eb69bfbf5d308b658f688d90237f50442db91b23cc8478d34ffe5c4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bd906f4d99b9b6041d9ea6d074ed9dcf3b6c3c6726310527a35162d819c37bb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6ba6e6f2acc5d13553b2e358a4299ae6a45c0a97799a20a4d2dafd5f08f357dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e21f96af2333400f525f12e16488a54d638fc04086494fc5e98f22e6e82e8407", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "89b874b0d4746bb787f1089c7b3ab5f8003f9432a0395e41e226b5e828bb2447", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0f29a3140a355d56c72687b4344d769251ff0bdf67a11710bf5e94f245ac7517", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1de8448be0d8663d66764f716bc53de96c085e50e17f8cb7cf5990cfbb07e7aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "eeb88651295cd1f9aa707a998c7784722648123c1e65fcbf05a9c2a1781d1c3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "96a6a70cedbcb270e2958a4a44fd9261ab65021826720e7ba4942b427df86996", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7c662653ad00292d5fc681e6fa35a7cff0ef8f83243b4e7a28a32e202346a0fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1a8e86e2bfeb0f42b71dbce1564ed7cc081b9b9362c2f1c57af7a604e0ff9dcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4d676cd5ef88f1567a145a263e87f2b82b5a7f625d166d3041f066344b953de4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b2ea45103d95b060be29c4f42a218535164876d9b96c0bc5ae1531d3336f44b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "490e57b31f6375314de8eb1761638ff8e05ccd006b18bac3011176c4d13d5f78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6d209ac99b25a8077fb09cc4e07aebf3f609175c9e37d602322a70131dda5473", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "82ff249f6a81a77f62b70535747f6950ed9a12279065ae25a9a0da2a2f79000d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c9b38120adb8308d01c5653b6a0903c5b648bd7f7cf33c394ae8a124a82e01a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9b2da306eae83f1131e0a094e96488a563f2c0f6abeeb8482e960feef472d73b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cb91a87c963bfd506e5fca4fe4bdf1f239e1a2052dbf3ddfeb6398b90ca83edf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9baac109a3f8808b4f0610f728c957038d3eab7a87bb32e7379a9c1763742700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0e31d2c822460a960d6e65b108804dcbf98c11e278f946a59d3e4d9b3a3e3990", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "de88c55d776c18f2d00d5dc051b56388cdb1a98f55dd4078ae7069689419079e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "104372c1eeca24209ae80ab6889751cbe2d74d906237f41385914037da473f1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1b09134f5ed494d66018c51cdea02c59a0584f7e63e4b49d70bc501b0be66950", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4c6159a141016b2d86e48842894f2f7dd5cde9afbd980324c9f6ed493f5e5226", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a511760c8b7f990354ff3a95e538ccd3bedd4a77e20bda2d28851e9c8285942d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4a27ed329ee074ef47b431054e01405beb323d2901f522daa8773725a51d2d97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d38759ceaa72b553b2da0af24b2b2e4b4689db9041443f773e03d63141a7d710", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "54cdfc00634451db788487373e3713c07dd1a3ad327a79c0f7657be2c4ae9bf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "de0be86ae105bf7bc27d4668fb35d0af8047885ac6df16f2bdaa60b358582cde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3bdd48d30e7e75a1ac8fbabb79bdcc3401ffd8f950449788d89645f7b5890c8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8eee8841ad84a5110d8ee6d72f86e8097409154483084e9a6bf2b8052de7043", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9ffb65a12725acddba1375ddc0d855f9c2c2db43b87ccfd6e2e097bbb0ab33ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ebd5a68f6742ba58b2256465db5a0704a639588d979ec24c9316a528e9135552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fe83844f089e5dcc11d852d3eb337026d28f0bfa77ded762e3ae3238997a4f7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1f2d066e3cb4d284dc8c41a3563538a18ca2f5e461a3f8ddc91edb092aba5e21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8f2c0d869f27fa4cfd0c7341c0a781decd21f4902172b797e6e8ddb2605b331f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c52bd24ee36725ec83c2685faae282b7588f2074018b0485f55ab4411b715e91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "825c0b30a3d6eee6594ea81aa9b90097c28af454e16dfb1b2e9736d4b21d95bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4995ff2a7bc858a940a8f9de36618dd0d381aa03da5da1a17b917721898e8ef9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "11af3e3bd7982aba3517a4a5d6d6de0554580d29338086b2fcba16e24fad7b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "59ca5026b90824b69bc642c7af964a762d5869c9c4a357aab0ec20042d9d3042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2eb7bcbaf7015c47d0bf195c1b27ddb9acb92212ca80a477409514ea3d55e41f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ceac3dfecbc8d39171bc5dbc94133859de00891459784c8aaa26bf5830071a0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "776b6517ea9706e0eac3039a4b87b7b7c4f693cd12b294033af47a52f76b3f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "71ba0187dbe828ce95c28e2828f92cc17279b4feb09409054e7b8f2b7d2e05a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "80d63d9b8a6109add1150f0fa0677d6992f040746a07b64aedc74fb25adf277b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c709a3902966515847c11c14a4fd56595a6f324b13e5600f527733cf6c9a9823", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "25f43a98adf85707db22e05e1b7d423e3e71e051f4e9e944f04b1de14e3e0093", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "363cf184aab3d5a4a8ff8cabf93cfd982b2e1b91b2b841268fcea7183026e69e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ceca22751d128b20841f29bc019b797af827db741b092c817062ed45a6c76af6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4304116d008f358d0864c25a93bc8c353c1406261735f62d010daf731f57bb61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3864780c03af24b9d8bd2512905663dbecd75fecef2c63e994ca99e56075abf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e6711e3c40bf68f9326de7b85a60b41804719ab43c5efb191a5ee49c4928e074", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e3574d7a199eb88654dfc18237986f6acf2f62b600ca12849fe7118d909a7663", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "95e2967c4e0f23e6912b923cf1c0ae46f150f603cfc32470c0b6a40b6989245c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ed8ea87a3890a8f1c5dd0e48f80f72aee4f0ab2c43ba8f138acb2b5de717e4ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "07c8f8bb87abf9ceed3dded48f630001fb8c768d48783bdd3cc522ecb86e21f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9ef526d71e512bcd4f6092ad805684062718a7fd52447db5a0b85963f567e49d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "7dcad61a6d89afc5ce1e377312b7621d36f637db97035ecf7bea64257d89e1f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ff4369ac2ac99b7ef588052aa289379eaf70fa68684d989c8905d791eb23c073", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8f0eb3cb7aef2b4e57de596fd635c9b626d85d3893a0513e6ac39210b2dd7be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "280a91d026e28e4f365a0c81a4087c14f091a50fbefa49bf05e140d5a0e74620", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e24316e35215eac099526c7cfa4333f5df4ccb6fb27ab7ca0d67d3e4c13c5638", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ca931a33aa47c108b89e5632a6406ceb6b25241624ad5fe5320e52cc92d54c1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b1aade2b8269ebe46d4dafd41836b7940cd0a17b581d74c7f43c18c791509c32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "da8b95138c1e65ac249eb1691eede1fea1960f2c92fed879afd854fc4bf791fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e6a6d321e2a62b97757fb807d498ed27cca7e8cf412234e25fe4b334fdac712b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "b60b146882af3146c01bfa301dcb7a1faa0612587a6936ae70bcc0036a073565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6b54cb1121f0a7ab72b1f52e130c741979b63280a04b95129789b5031e386c81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "20cbb898c0bb32902afd86f47c38d9ed45df74cda5263c25cc5addd0bf288dd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "04d94aa72f155f6e85eb73c00da6e4121dd227496366cd5f944f91aa0943b6ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "a7b994988f61497ff8f885c748f49351bc84c3ba393c259c856bc3dd15741dff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "4d560d1f1a369eb476172dc2dca165a650e63c38954ff34b413cac6f5db09957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "71b124a071a2a32dcd9467ade0fd78ddbc3e498fe24a0aecdf01cfedd47853c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c0655d77589b0970144166df73c67317d41ce65f403867a82f7bb38b4020e490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "b88d49cbf54b651e5fc1ed0c5e2e06c80a1eca07f36ebac34fa19bcb49c1de30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c11a5159d3709d694157e9bfaae04bb7fa74d9c880fbf99db58b0fb31950a269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "00d2081769a912a993816cab2c0b67c49ab93868e3cb75354f2cd03c8bc8b65a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "0ea541a68b2d955202226e64aed5a42ced6740e57fba46845af4ae267caca26e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "8a23887da538f5c924d442b59ef8eca4a8b5927e8ea9525c6fc45e8587a6305f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "61ff8b7c2a5e281b852995f7f9fd7920507c5a3bc5e1bae83e765045582a50a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c3d5c741a5567a2e1f80e2838cfdd2b0248d22f764bf160dad53bac4ac268a1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "4da0041c350a1ace7022d4bf9837b501c3b8b312b601ed55e23d7e28bea6cfce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "824348525259965833c1140c26db5ab11728e5d3410c793caa8ed4cd3e0c69ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "adbc2bb6900b50f3bb20e1e7cb65a28c58a060e94b0699e8d87fcf4150845b23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "2c7b06ce177f25e6c51ba4b0386a7b8cd70b902cf92aa103927bcb0b56265410", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "719d29fa46812ef6025de470f5b3868a0cb2b7a5533f0dda9c9bcffa221732e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6c8fbf9fbd12d3eb2a568ecf0d64acf96e87b6d83c42db625030b4bdb13ca528", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEnablesGlobalPtxKnobs */ 1
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "72dccde2ceaa5a5d971e5faa2f5959acf7c8169738a4caf500cc3cc70d30efd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "fc2e05b5069df8b4ed02fc45b8ffcfa3103e721a130693e3a8ec241f5e363f2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "8f1fd154608e866412c3648b0db616ff14847ea954bcab94fc386471bf92bbae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "fe4a6be2ff13e3b16e2f163f1aaff8d46590a9a7a573d374c388f9a759b52b39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "2c84e5fa2d73c125291ba46d8fe8d2430b914ffdf46ffb42c7c4d1662879f9f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "d5f4ff363008d1e59afd011ce4e6ea7e1e8cf593c6bc78b93f9daf912d4ee51f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "268d6a7af220c1bc094288aeac682521e920fb1c33f8705d4197a439247a492e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* 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 */ {}
, /* mLamportConsumerA */ 0
, /* mLamportConsumerB */ 0
, /* mLamportForceValid */ 0
, /* mLamportProducer */ 0
, /* 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
, /* mPatchPzo */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(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
, /* mUseUniqueSfA */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
#endif // EXCLUDE_SM_100
};
// clang-format on
} // namespace kernels
} // namespace tensorrt_llm

}

